Update app.py

app.py

@@ -30,9 +30,9 @@ def inference(input_text):
     predicted_class_id = logits.argmax().item()
     output = model.config.id2label[predicted_class_id]
     ans = "积极"
-    # if output == "0":
-    #     ans = "消极"
-    return output
+    if output == "0":
+        ans = "消极"
+    return ans
 
 demo = gr.Interface(
     fn=inference,