text-classifier-project1

Sleeping

sitammeur commited on Jan 11

Commit

66a505e

verified ·

1 Parent(s): 5ba7182

Update src/app/predict.py

Files changed (1) hide show

src/app/predict.py CHANGED Viewed

@@ -38,7 +38,7 @@ def ZeroShotTextClassification(
             text_input,
             labels,
             hypothesis_template=hypothesis_template,
-            multi_label=True,
         )
         # Return the classification results

             text_input,
             labels,
             hypothesis_template=hypothesis_template,
+            multi_label=False,
         )
         # Return the classification results