Hugging Face's logo

Spaces:
peter2000
/
vulnerable_groups
Sleeping

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -37,8 +37,7 @@ group_dict = {
37
  29: 'Migrants and displaced populations'}
38
 
39
  def predict(text):
40
- preds = model(text)[0].item()
41
- return preds
42
  return group_dict[preds]
43
 
44
  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(