Spaces:

hhhwmws
/

test_Idiot-Cultivation-System

Sleeping

hhhwmws commited on Aug 8

Commit

438c612

•

1 Parent(s): 56e6e25

Update src/ImageBase.py

Files changed (1) hide show

src/ImageBase.py CHANGED Viewed

@@ -36,7 +36,7 @@ class Imagebase:
             except:
                 from src.CLIPExtractor import CLIPExtractor
-            cache_dir = "D:\\aistudio\\LubaoGithub\\models"
             self.clip_extractor = CLIPExtractor(model_name="openai/clip-vit-large-patch14", cache_dir=cache_dir)
     def top_k_search(self, query_feature, top_k=15):

             except:
                 from src.CLIPExtractor import CLIPExtractor
+            cache_dir = "models"
             self.clip_extractor = CLIPExtractor(model_name="openai/clip-vit-large-patch14", cache_dir=cache_dir)
     def top_k_search(self, query_feature, top_k=15):