sohojoe commited on
Commit
a9409d4
1 Parent(s): e81e68c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -270,8 +270,7 @@ def on_example_image_click_set_image(input_image, image_url):
270
  # device = torch.device("mps" if torch.backends.mps.is_available() else "cuda:0" if torch.cuda.is_available() else "cpu")
271
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
272
 
273
- from clip_retrieval.load_clip import load_clip, get_tokenizer
274
- # model, preprocess = load_clip(clip_model, use_jit=True, device=device)
275
  # model, preprocess = load_clip(clip_model, use_jit=True, device=device)
276
  # tokenizer = get_tokenizer(clip_model)
277
  model, preprocess, tokenizer = None, None, None
 
270
  # device = torch.device("mps" if torch.backends.mps.is_available() else "cuda:0" if torch.cuda.is_available() else "cpu")
271
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
272
 
273
+ # from clip_retrieval.load_clip import load_clip, get_tokenizer
 
274
  # model, preprocess = load_clip(clip_model, use_jit=True, device=device)
275
  # tokenizer = get_tokenizer(clip_model)
276
  model, preprocess, tokenizer = None, None, None