Hugging Face's logo

Spaces:
peter2000
/
vulnerable_groups
Sleeping

App Files Community
peter2000 commited on
Commit
4e389ed
·
1 Parent(s): 79a6f1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ group_dict = {
37
  29: 'Migrants and displaced populations'}
38
 
39
  def predict(text):
40
- preds = model(text)[0].item())
41
  return group_dict[preds]
42
 
43
  gradio_ui = gr.Interface(
 
37
  29: 'Migrants and displaced populations'}
38
 
39
  def predict(text):
40
+ preds = model([text])[0].item()
41
  return group_dict[preds]
42
 
43
  gradio_ui = gr.Interface(