Update app.py

app.py

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