Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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():
|