Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -60,9 +60,8 @@ def generate_image_embeddings(prompt_image,
|
|
60 |
projector,
|
61 |
device='cuda:0'):
|
62 |
prompt_image = prompt_image.convert('RGB')
|
63 |
-
inputs = vision_processor(images=
|
64 |
-
return_tensors="pt"
|
65 |
-
padding=True)
|
66 |
inputs = inputs.to(device)
|
67 |
image_outputs = vision_encoder(**inputs)
|
68 |
img_feats = image_outputs.image_embeds.view(1, -1)
|
|
|
60 |
projector,
|
61 |
device='cuda:0'):
|
62 |
prompt_image = prompt_image.convert('RGB')
|
63 |
+
inputs = vision_processor(images=prompt_image,
|
64 |
+
return_tensors="pt")
|
|
|
65 |
inputs = inputs.to(device)
|
66 |
image_outputs = vision_encoder(**inputs)
|
67 |
img_feats = image_outputs.image_embeds.view(1, -1)
|