Update pipeline.py

pipeline.py

@@ -3,8 +3,6 @@ from typing import Any, Dict, List
 
 import tensorflow as tf
 from tensorflow import keras
-from app.pipelines import Pipeline
-from huggingface_hub import from_pretrained_keras, hf_hub_download
 from PIL import Image
 import base64
 
@@ -23,15 +21,7 @@ class PreTrainedPipeline(Pipeline):
         # Number of labels
         self.num_labels = self.model.output_shape[1]
 
-        # Config is required to know the mapping to label.
-        #config_file = hf_hub_download(model_id, filename=CONFIG_FILENAME)
-        #with open(config_file) as config:
-        #    config = json.load(config)
-        
-        self.num_labels = 3
-        self.id2label = config.get(
-            "id2label", {str(i): f"LABEL_{i}" for i in range(self.num_labels)}
-        )
+        self.id2label = self.id2label = {"0": "pet", "1":"no_pet"}
 
     def __call__(self, inputs: "Image.Image") -> List[Dict[str, Any]]:
         """