Update app.py

app.py

@@ -33,7 +33,7 @@ extract_path = "sample_evaluation"
 unzip_file(zip_path, extract_path)
 
 from huggingface_hub import hf_hub_download
-hf_hub_download(repo_id="safinal/compositional-retrieval", filename="weights.pth", local_dir='.')
+hf_hub_download(repo_id="safinal/compositional-image-retrieval", filename="weights.pth", local_dir='.')
 
 
 
@@ -75,7 +75,7 @@ def process_single_query(model, query_image_path, query_text, database_embedding
     
     # Get token classifier
     token_classifier, token_classifier_tokenizer = load_token_classifier(
-        "safinal/compositional-retrieval-token-classifier",
+        "safinal/compositional-image-retrieval-token-classifier",
         device
     )