Hugging Face's logo

Spaces:
sourabhbargi11
/
caption_generate
Sleeping

App Files Community
sourabhbargi11 commited on
Commit
94fd841
·
verified ·
1 Parent(s): c73056a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def image_preprocess(image):
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
37
  with torch.no_grad():
 
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
37
  with torch.no_grad():