Spaces:
Build error
Build error
mnavas
commited on
Commit
·
81a59ae
1
Parent(s):
06b18ea
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 |
|