Update app.py

app.py

@@ -68,17 +68,17 @@ def classify_text(text):
 
     inputs = tokenizer(cleaned_text, return_tensors="pt", truncation=True, padding=True).to(device)
 
-        with torch.no_grad():
-            logits_1 = model_1(**inputs).logits
-            logits_2 = model_2(**inputs).logits
-            logits_3 = model_3(**inputs).logits
+    with torch.no_grad():
+        logits_1 = model_1(**inputs).logits
+        logits_2 = model_2(**inputs).logits
+        logits_3 = model_3(**inputs).logits
     
-            softmax_1 = torch.softmax(logits_1, dim=1)
-            softmax_2 = torch.softmax(logits_2, dim=1)
-            softmax_3 = torch.softmax(logits_3, dim=1)
+        softmax_1 = torch.softmax(logits_1, dim=1)
+        softmax_2 = torch.softmax(logits_2, dim=1)
+        softmax_3 = torch.softmax(logits_3, dim=1)
     
-            averaged_probabilities = (softmax_1 + softmax_2 + softmax_3) / 3
-            probabilities = averaged_probabilities[0]
+        averaged_probabilities = (softmax_1 + softmax_2 + softmax_3) / 3
+        probabilities = averaged_probabilities[0]
 
     ai_probs = probabilities.clone()
     ai_probs[24] = 0