mnavas commited on
Commit
06b18ea
·
1 Parent(s): d77da02
Files changed (1) hide show
  1. app.py +1 -0
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()