Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
178625c
1
Parent(s):
8c7ce11
Update app.py
Browse files
app.py
CHANGED
@@ -109,7 +109,7 @@ def inference(text,steps,image):
|
|
109 |
all_frames = []
|
110 |
target = clip_model.embed_text(text)
|
111 |
if image:
|
112 |
-
target = embed_image(TF.to_tensor(image
|
113 |
else:
|
114 |
target = clip_model.embed_text(text)
|
115 |
steps = steps
|
|
|
109 |
all_frames = []
|
110 |
target = clip_model.embed_text(text)
|
111 |
if image:
|
112 |
+
target = embed_image(TF.to_tensor(image).to(device).unsqueeze(0)).mean(0).squeeze(0)
|
113 |
else:
|
114 |
target = clip_model.embed_text(text)
|
115 |
steps = steps
|