Update app.py

app.py

@@ -39,7 +39,7 @@ group_dict = {
 
 # Define prediction function 
 def predict(text):
-    preds = model([text])[0].item()
+    preds = model(text)
     return group_dict[preds]
 
 # App