Spaces:
Build error
Build error
mnavas
commited on
Commit
·
06b18ea
1
Parent(s):
d77da02
app.py
CHANGED
@@ -28,6 +28,7 @@ def askcpv(description):
|
|
28 |
# turn predicted id's into actual label names
|
29 |
# predicted_labels = [id2label[idx] for idx, label in enumerate(predictions) if label == 1.0]
|
30 |
# return predicted_labels
|
|
|
31 |
return {cpv[i]: v.item() for i, v in zip(indices, values)}
|
32 |
|
33 |
gr.Interface(fn=askcpv, inputs="textbox", outputs="label").launch()
|
|
|
28 |
# turn predicted id's into actual label names
|
29 |
# predicted_labels = [id2label[idx] for idx, label in enumerate(predictions) if label == 1.0]
|
30 |
# return predicted_labels
|
31 |
+
print({i: v.item() for i, v in zip(indices, values)})
|
32 |
return {cpv[i]: v.item() for i, v in zip(indices, values)}
|
33 |
|
34 |
gr.Interface(fn=askcpv, inputs="textbox", outputs="label").launch()
|