Update app.py

app.py

@@ -29,7 +29,7 @@ def predict_sentiment(model, input_tokens):
     output = model(**input_tokens).pooler_output.detach().numpy()
     with open('LogReg_imdb_Ernie.pkl', 'rb') as file:
         cls = pickle.load(file)
-    result = id2label[cls.predict(output).tolist()]
+    result = id2label[int(cls.predict(output))]
     return result
 
 st.title('Text sentiment analysis by ErnieModel')