Commit ·
d231d89
1
Parent(s): f27c209
Added self in the method call
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -66,9 +66,9 @@ class EndpointHandler():
|
|
| 66 |
# img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
|
| 67 |
# raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|
| 68 |
|
| 69 |
-
inputs = processor(raw_image, prompt, return_tensors="pt")
|
| 70 |
|
| 71 |
-
out = model.generate(**inputs)
|
| 72 |
captions = processor.decode(out[0], skip_special_tokens=True)
|
| 73 |
|
| 74 |
return {"captions": captions}
|
|
|
|
| 66 |
# img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'
|
| 67 |
# raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')
|
| 68 |
|
| 69 |
+
inputs = self.processor(raw_image, prompt, return_tensors="pt")
|
| 70 |
|
| 71 |
+
out = self.model.generate(**inputs)
|
| 72 |
captions = processor.decode(out[0], skip_special_tokens=True)
|
| 73 |
|
| 74 |
return {"captions": captions}
|