sg7849 commited on
Commit
1b1c5b7
·
verified ·
1 Parent(s): 07c9a27

key update

Browse files
Files changed (1) hide show
  1. hf_retrieval.py +2 -4
hf_retrieval.py CHANGED
@@ -15,11 +15,9 @@ checkpoint_path = "finetuned_clip.pt"
15
  model.load_state_dict(torch.load(checkpoint_path, map_location="cpu"))
16
  model.eval()
17
 
18
- print("Type of QDRANT_CLOUD_URL:", type(os.environ["QDRANT_CLOUD_URL"]))
19
- print("Type of QDRANT_API_KEY:", type(os.environ["QDRANT_API_KEY"]))
20
-
21
  qdrant = QdrantClient(url=os.environ["QDRANT_CLOUD_URL"],
22
- api_key=["QDRANT_API_KEY"])
 
23
  collection_name = "video_chunks"
24
 
25
 
 
15
  model.load_state_dict(torch.load(checkpoint_path, map_location="cpu"))
16
  model.eval()
17
 
 
 
 
18
  qdrant = QdrantClient(url=os.environ["QDRANT_CLOUD_URL"],
19
+ api_key=["QDRANT_API_KEY"],
20
+ prefer_grpc=False)
21
  collection_name = "video_chunks"
22
 
23