tombio commited on
Commit
defcb0d
1 Parent(s): 6390664

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,8 +37,8 @@ def get_url_im(t):
37
 
38
  @torch.no_grad()
39
  def get_im_c(im_path, clip_model):
40
- # im = Image.open(im_path).convert("RGB")
41
- prompts = preprocess(im_path).to(device).unsqueeze(0)
42
  return clip_model.encode_image(prompts).float()
43
 
44
  @torch.no_grad()
 
37
 
38
  @torch.no_grad()
39
  def get_im_c(im_path, clip_model):
40
+ im = Image.open(im_path).convert("RGB")
41
+ prompts = preprocess(im).to(device).unsqueeze(0)
42
  return clip_model.encode_image(prompts).float()
43
 
44
  @torch.no_grad()