sourabhbargi11 commited on
Commit
84fa46d
·
verified ·
1 Parent(s): 553f724

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ def image_preprocess(image):
32
 
33
  def generate_caption(image_processor, model, tokenizer,device, image):
34
  inputs = image_processor(image, return_tensors='pt').to(device)
 
35
  model.eval()
36
  # Generate caption
37
  with torch.no_grad():
 
32
 
33
  def generate_caption(image_processor, model, tokenizer,device, image):
34
  inputs = image_processor(image, return_tensors='pt').to(device)
35
+ print(inputs)
36
  model.eval()
37
  # Generate caption
38
  with torch.no_grad():