Modified req.txt and app.py

app.py

@@ -196,14 +196,14 @@ def get_matched_image(matches, val_file_dict_loaded):
 
 def get_grayscale_image(text_query):
     model_inf = CLIPModel().to('cpu')
-    model_inf.load_state_dict(torch.load('./best_clip_model_cpu.pt', map_location='cpu'))
+    model_inf.load_state_dict(torch.load('best_clip_model_cpu.pt', map_location='cpu'))
 
-    clip_image_embeddings_np_inf = np.load('./clip_image_embeddings.npy')
+    clip_image_embeddings_np_inf = np.load('clip_image_embeddings.npy')
     image_embeddings_inf = torch.tensor(clip_image_embeddings_np_inf)
 
-    img_file_names = np.load('./val_img_file_names.npy',allow_pickle=True)
+    img_file_names = np.load('val_img_file_names.npy',allow_pickle=True)
 
-    with open("./val_imgs_rle_encode.json", "r") as json_file:
+    with open("val_imgs_rle_encode.json", "r") as json_file:
         val_file_dict_loaded = json.load(json_file)
 
     matches = find_matches_cpu(model_inf,

requirements.txt

@@ -1,4 +1,7 @@
-transformers
-gradio
-timm
-torch
+transformers==4.33.3
+gradio==3.46.0
+timm==0.9.7
+--extra-index-url https://download.pytorch.org/whl/cpu/
+torch==2.0.0+cpu
+torchvision==0.15.1+cpu
+opencv-python-headless==4.8.0.76