Update app.py

app.py

@@ -17,11 +17,11 @@ def get_output(text):
     model_output = model(**model_input, return_dict=False)
     prediction = torch.argmax(model_output[0].cpu(), dim=-1)
     prediction = [p.item() for p in prediction]
-    # for i in range(len(prediction)):
-    #     if prediction[i]==0:
-    #         output.append("消极")
-    #     else:
-    #         output.append('积极')
+    for i in range(len(prediction)):
+        if prediction[i]==1:
+            output.append("消极")
+        else:
+            output.append('积极')