Removed hardcoded url

handler.py

@@ -12,7 +12,7 @@ class EndpointHandler:
         self.tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        parameters = data.pop("parameters", {"mode": "image"})
+        parameters = data.pop("parameters", {"mode": "text"})
         inputs = data.pop("inputs", data)
         with torch.no_grad():
             if parameters["mode"] == "text":
@@ -20,8 +20,7 @@ class EndpointHandler:
                 features = self.model.get_text_features(**inputs)
 
             if parameters["mode"] == "image":
-                url = "http://images.cocodataset.org/val2017/000000039769.jpg"
-                image = Image.open(requests.get(url, stream=True).raw)
+                image = Image.open(requests.get(inputs, stream=True).raw)
 
                 inputs = self.processor(images=image, return_tensors="pt")
                 features = self.model.get_image_features(**inputs)