TheoLvs commited on
Commit
07f05b1
·
1 Parent(s): f711847

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,9 +7,9 @@ model = SetFitModel.from_pretrained("eclaircies/ecolo-pas-ecolo-v0.1")
7
  # Labels
8
  labels = ['pas écolo', 'ça dépend', 'écolo']
9
 
10
- def greet(name):
11
- preds = int(model([x]).detach().numpy())
12
  return labels[preds]
13
 
14
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
15
  iface.launch()
 
7
  # Labels
8
  labels = ['pas écolo', 'ça dépend', 'écolo']
9
 
10
+ def predict(concept):
11
+ preds = int(model([concept]).detach().numpy())
12
  return labels[preds]
13
 
14
+ iface = gr.Interface(fn=predict, inputs="text", outputs="text")
15
  iface.launch()