Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -146,7 +146,7 @@ def predict1(image_input, question):
|
|
| 146 |
# Pass the image and encoded prompt to the model
|
| 147 |
with torch.no_grad():
|
| 148 |
# Get image embeddings
|
| 149 |
-
image_embeddings =
|
| 150 |
projected_image_embeddings = model.image_projection(image_embeddings)
|
| 151 |
|
| 152 |
# Reshape image embeddings to (batch_size, 1, phi3_embed_dim)
|
|
|
|
| 146 |
# Pass the image and encoded prompt to the model
|
| 147 |
with torch.no_grad():
|
| 148 |
# Get image embeddings
|
| 149 |
+
image_embeddings = image_encoder(image)
|
| 150 |
projected_image_embeddings = model.image_projection(image_embeddings)
|
| 151 |
|
| 152 |
# Reshape image embeddings to (batch_size, 1, phi3_embed_dim)
|