Update Dockerfile

Dockerfile

@@ -10,25 +10,4 @@ RUN apt-get -y update && \
 COPY requirements.txt .
 RUN pip install -r requirements.txt
 
-class ZeroShotTextClassifier:
-  # Class variable for the model
-  classifier = None
-  @classmethod
-  def load(cls):
-    if cls.classifier is None:
-      # Load the model only once
-      cls.classifier = pipeline("zero-shot-classification", 
-                         model="facebook/bart-large-mnli")
-  @classmethod
-  def predict(cls, text, candidate_labels):
-    # Ensure the model is loaded
-    cls.load()
-    # Predict
-    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]
-    return {"label": label, "score": score}
-
 RUN python -c "from transformers import pipeline; classifier = pipeline('zero-shot-classification', model='facebook/bart-large-mnli')"