sohojoe commited on
Commit
c71f2a1
1 Parent(s): f3db7a5

Revert "Update app.py"

Browse files

This reverts commit fca5b2fd7d7598b474cf4a2a46d2a85aee9563d0.

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,6 +9,8 @@ import math
9
  # from transformers import CLIPTextModel, CLIPTokenizer
10
  import os
11
 
 
 
12
 
13
  # clip_model_id = "openai/clip-vit-large-patch14-336"
14
  # clip_retrieval_indice_name, clip_model_id ="laion5B-L-14", "/laion/CLIP-ViT-L-14-laion2B-s32B-b82K"
@@ -269,7 +271,6 @@ def on_example_image_click_set_image(input_image, image_url):
269
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
270
 
271
  # from clip_retrieval.load_clip import load_clip, get_tokenizer
272
- # from clip_retrieval.clip_client import ClipClient, Modality
273
  # model, preprocess = load_clip(clip_model, use_jit=True, device=device)
274
  # tokenizer = get_tokenizer(clip_model)
275
  # clip_retrieval_client = ClipClient(
 
9
  # from transformers import CLIPTextModel, CLIPTokenizer
10
  import os
11
 
12
+ from clip_retrieval.clip_client import ClipClient, Modality
13
+
14
 
15
  # clip_model_id = "openai/clip-vit-large-patch14-336"
16
  # clip_retrieval_indice_name, clip_model_id ="laion5B-L-14", "/laion/CLIP-ViT-L-14-laion2B-s32B-b82K"
 
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
  # clip_retrieval_client = ClipClient(