sourabhbargi11 commited on
Commit
628a420
·
verified ·
1 Parent(s): 94fd841

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def image_preprocess(image):
30
  image = image.convert("RGB")
31
  return image
32
 
33
- def generate_caption(processor, model, device, image):
34
  inputs = image_processor(image, return_tensors='pt').to(device)
35
  model.eval()
36
  # Generate caption
 
30
  image = image.convert("RGB")
31
  return image
32
 
33
+ def generate_caption(image_processor, model, device, image):
34
  inputs = image_processor(image, return_tensors='pt').to(device)
35
  model.eval()
36
  # Generate caption