RobotJelly commited on
Commit
4640c21
·
1 Parent(s): b2aa59f
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,7 +9,7 @@ from io import BytesIO
9
  import requests
10
  import gradio as gr
11
  import os
12
- from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
13
  import sentence_transformers
14
  from sentence_transformers import SentenceTransformer, util
15
 
@@ -26,12 +26,12 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
26
  #photo_ids = list(photo_ids['photo_id'])
27
 
28
  # Photo dataset
29
- photos = pd.read_csv("./photos.tsv000", sep="\t", header=0)
30
 
31
  # taking features vectors
32
- photo_features = np.load("./features.npy")
33
 
34
- IMAGES_DIR = './photos'
35
 
36
 
37
 
 
9
  import requests
10
  import gradio as gr
11
  import os
12
+ #from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
13
  import sentence_transformers
14
  from sentence_transformers import SentenceTransformer, util
15
 
 
26
  #photo_ids = list(photo_ids['photo_id'])
27
 
28
  # Photo dataset
29
+ #photos = pd.read_csv("./photos.tsv000", sep="\t", header=0)
30
 
31
  # taking features vectors
32
+ #photo_features = np.load("./features.npy")
33
 
34
+ #IMAGES_DIR = './photos'
35
 
36
 
37