Update app.py

app.py

@@ -37,7 +37,7 @@ group_dict = {
  29: 'Migrants and displaced populations'}
 
 def predict(text):
-    preds = model(text)[0].item())
+    preds = model([text])[0].item()
     return group_dict[preds]
 
 gradio_ui = gr.Interface(