sitammeur commited on
Commit
b256293
·
verified ·
1 Parent(s): 2ce26a5

Update classifier.py

Browse files
Files changed (1) hide show
  1. 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)}