Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
11 |
-
preds = int(model([
|
12 |
return labels[preds]
|
13 |
|
14 |
-
iface = gr.Interface(fn=
|
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()
|