Hugging Face's logo

Spaces:
hhhwmws
/
test_Idiot-Cultivation-System
Sleeping

App Files Community
hhhwmws commited on
Commit
438c612
·
verified ·
1 Parent(s): 56e6e25

Update src/ImageBase.py

Browse files
Files changed (1) hide show
  1. src/ImageBase.py +1 -1
src/ImageBase.py CHANGED
@@ -36,7 +36,7 @@ class Imagebase:
36
  except:
37
  from src.CLIPExtractor import CLIPExtractor
38
 
39
- cache_dir = "D:\\aistudio\\LubaoGithub\\models"
40
  self.clip_extractor = CLIPExtractor(model_name="openai/clip-vit-large-patch14", cache_dir=cache_dir)
41
 
42
  def top_k_search(self, query_feature, top_k=15):
 
36
  except:
37
  from src.CLIPExtractor import CLIPExtractor
38
 
39
+ cache_dir = "models"
40
  self.clip_extractor = CLIPExtractor(model_name="openai/clip-vit-large-patch14", cache_dir=cache_dir)
41
 
42
  def top_k_search(self, query_feature, top_k=15):