hughtayloe
commited on
Commit
•
c8ebd57
1
Parent(s):
974fbaa
Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -24,5 +24,5 @@ class EndpointHandler():
|
|
24 |
raw_image = Image.open(requests.get(url, stream=True).raw)
|
25 |
inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
|
26 |
output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
|
27 |
-
|
28 |
-
return
|
|
|
24 |
raw_image = Image.open(requests.get(url, stream=True).raw)
|
25 |
inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
|
26 |
output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
|
27 |
+
readable = (self.processor.decode(output[0][2:], skip_special_tokens=True))
|
28 |
+
return readable
|