Update Dockerfile

Dockerfile

@@ -27,8 +27,8 @@ class ZeroShotTextClassifier:
     huggingface_predictions = cls.classifier(text, candidate_labels)
     # Create our own prediction object with the best label
     max_index = np.argmax(huggingface_predictions["scores"])
-    label = huggingface_predictions["labels"][max_index]
-    score = huggingface_predictions["scores"][max_index]
+    label = huggingface_predictions['labels'][max_index]
+    score = huggingface_predictions['scores'][max_index]
     return {"label": label, "score": score}
 
 RUN python -c "from transformers import pipeline; classifier = pipeline('zero-shot-classification', model='facebook/bart-large-mnli')"