hughtayloe commited on
Commit
b9c7f3e
·
verified ·
1 Parent(s): 1ac6526

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -35,6 +35,6 @@ class EndpointHandler():
35
  raw_image = Image.open(requests.get(link, stream=True).raw)
36
  inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
37
  output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
38
- print(self.processor.decode(output[0][2:], skip_special_tokens=True))
39
- outputs.append(url)
40
  return outputs
 
35
  raw_image = Image.open(requests.get(link, stream=True).raw)
36
  inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
37
  output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
38
+ readable = (self.processor.decode(output[0][2:], skip_special_tokens=True))
39
+ outputs.append(readable)
40
  return outputs