Update app.py
Browse files
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
|