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, 2024

Commit

9c503ae

·

1 Parent(s): d9d7f33

look for inputs

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -19,6 +19,7 @@ class EndpointHandler():
         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)

         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)
+        inputs = data.pop("inputs", None)
         image = Image.open(requests.get(url, stream=True).raw)