rishabh-zuma commited on
Commit
d231d89
·
1 Parent(s): f27c209

Added self in the method call

Browse files
Files changed (1) hide show
  1. 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}