fairewholesale
/

product-taxonomy-llava-hf-v1.6-13b

Image-Text-to-Text

Inference Endpoints

Model card Files Files and versions Community

hongyiyang commited on Jul 16

Commit

d9d7f33

•

1 Parent(s): 680f9d0

minor fix

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -18,10 +18,10 @@ class EndpointHandler():
         # get inputs
         prompt = "USER: <image>\nWhat's in the image\nASSISTANT:"
         default_url = "https://cdn.faire.com/fastly/3c335e5c06d3027964ee8351093784c94dfa264e5eb26430c803f4ab3c44da84.jpeg"
         image = Image.open(requests.get(url, stream=True).raw)
-        url = data.pop("image_url", default_url)
         # run normal prediction
         outputs = self.pipeline(image, prompt=prompt, generate_kwargs={"max_new_tokens": 200})

         # get inputs
         prompt = "USER: <image>\nWhat's in the image\nASSISTANT:"
         default_url = "https://cdn.faire.com/fastly/3c335e5c06d3027964ee8351093784c94dfa264e5eb26430c803f4ab3c44da84.jpeg"
+        url = data.pop("image_url", default_url)
         image = Image.open(requests.get(url, stream=True).raw)
         # run normal prediction
         outputs = self.pipeline(image, prompt=prompt, generate_kwargs={"max_new_tokens": 200})