Ahsen Khaliq commited on
Commit
178625c
1 Parent(s): 8c7ce11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.name).to(device).unsqueeze(0)).mean(0).squeeze(0)
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