Update app.py

app.py

@@ -22,7 +22,7 @@ def classify_text(text):
     # Get the probability of the class '1'
     prob_1 = probabilities[0][1].item()
     
-    return {"Probability of being 1": prob_1}
+    return {"Probability of being AI": prob_1}
 
 # Create the Gradio interface
 iface = gr.Interface(