Update app.py

app.py

@@ -17,15 +17,15 @@ 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]==0:
+    #         output.append("消极")
+    #     else:
+    #         output.append('积极')
 
 
 
-    return output
+    return prediction
 
 demo=gr.Interface(fn=get_output,inputs='text',outputs='text')
 demo.launch()