debugging 1

app.py

@@ -9,9 +9,9 @@ chatgpt_roberta_detector = pipeline("text-classification", model="Models/fine_tu
 
 def classify_text(text):
     # Get predictions from each model
-    roberta_base_pred = np.argmax(roberta_base_detector(text)[0]['label'].to('cpu'))
-    chatgpt_lli_hc3_pred = np.argmax(chatgpt_lli_hc3_detector(text)[0]['label'].to('cpu'))
-    chatgpt_roberta_pred = np.argmax(chatgpt_roberta_detector(text)[0]['label'].to('cpu'))
+    roberta_base_pred = roberta_base_detector(text)[0]['label']
+    chatgpt_lli_hc3_pred = chatgpt_lli_hc3_detector(text)[0]['label']
+    chatgpt_roberta_pred = chatgpt_roberta_detector(text)[0]['label']
 
     # Count the votes for AI and Human
     votes = {"AI": 0, "Human": 0}
@@ -23,9 +23,9 @@ def classify_text(text):
 
     # Determine final decision based on majority
     if votes["AI"] > votes["Human"]:
-        return "AI"
+        return roberta_base_pred
     else:
-        return "Human"
+        return roberta_base_pred
 
 # Create Gradio Interface
 iface = gr.Interface(