pamunarr commited on
Commit
65d794f
·
verified ·
1 Parent(s): 749f8db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ learner = from_pretrained_fastai(repo_id)
7
  labels = ["World" , "Nigeria" , "Health" , "Africa" , "Politics"]
8
 
9
  def predict(text):
10
- _ , _ , probs = learner.predict(img)
11
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
12
 
13
  gr.Interface(fn=predict, inputs="text", outputs=gr.components.Label(num_top_classes=5)).launch(share=False)
 
7
  labels = ["World" , "Nigeria" , "Health" , "Africa" , "Politics"]
8
 
9
  def predict(text):
10
+ _ , _ , probs = learner.predict(text)
11
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
12
 
13
  gr.Interface(fn=predict, inputs="text", outputs=gr.components.Label(num_top_classes=5)).launch(share=False)