wishmi1234 commited on
Commit
bcc8cce
·
verified ·
1 Parent(s): 607b9e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ class ImageCaptioningTool(Tool):
64
  device=0 if torch.cuda.is_available() else -1
65
  )
66
 
67
- def use(self, image, question):
68
  if not isinstance(image, Image.Image):
69
  image = Image.open(BytesIO(image))
70
  result = self.captioner(image)
 
64
  device=0 if torch.cuda.is_available() else -1
65
  )
66
 
67
+ def forward(self, image, question):
68
  if not isinstance(image, Image.Image):
69
  image = Image.open(BytesIO(image))
70
  result = self.captioner(image)