second model ensemble

app.py

@@ -39,8 +39,11 @@ def classify_text(text):
         logits_1 = model_1(**inputs).logits
         logits_2 = model_2(**inputs).logits
 
-        avg_logits = (logits_1 + logits_2) / 2
-        probabilities = torch.softmax(avg_logits, dim=1)[0]
+        softmax_1 = torch.softmax(logits_1, dim=1)
+        softmax_2 = torch.softmax(logits_2, dim=1)
+
+        averaged_probabilities = (softmax_1 + softmax_2) / 2
+        probabilities = averaged_probabilities[0]
 
     ai_probs = probabilities.clone()
     ai_probs[24] = 0
@@ -63,6 +66,9 @@ def classify_text(text):
     return result_message
 
 
+
+
+
 title = "AI Text Detector"
 
 description = """