Spaces:
Sleeping
Sleeping
Update classifier.py
Browse files- classifier.py +0 -1
classifier.py
CHANGED
|
@@ -60,5 +60,4 @@ class MedSigLIPClassifier:
|
|
| 60 |
|
| 61 |
# Format results
|
| 62 |
probs_list = probs[0].tolist()
|
| 63 |
-
print({label: prob for label, prob in zip(candidate_labels, probs_list)})
|
| 64 |
return {label: prob for label, prob in zip(candidate_labels, probs_list)}
|
|
|
|
| 60 |
|
| 61 |
# Format results
|
| 62 |
probs_list = probs[0].tolist()
|
|
|
|
| 63 |
return {label: prob for label, prob in zip(candidate_labels, probs_list)}
|