mnavas commited on
Commit
81a59ae
·
1 Parent(s): 06b18ea
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -28,6 +28,9 @@ 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
  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
 
 
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("probs: ", probs)
32
+ print("values: ", values)
33
+ print("indices: ", indices)
34
  print({i: v.item() for i, v in zip(indices, values)})
35
  return {cpv[i]: v.item() for i, v in zip(indices, values)}
36