Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def image_preprocess(image):
|
|
30 |
image = image.convert("RGB")
|
31 |
return image
|
32 |
|
33 |
-
def generate_caption(
|
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
|