Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
-os.system("gdown https://drive.google.com/uc?id=14pXWwB4Zm82rsDdvbGguLfx9F8aM7ovT")
-os.system("gdown https://drive.google.com/uc?id=1IdaBtMSvtyzF0ByVaBHtvM0JYSXRExRX")
+from huggingface_hub import hf_hub_download
+conceptual_weight = hf_hub_download(repo_id="akhaliq/CLIP-prefix-captioning-conceptual-weights", filename="conceptual_weights.pt")
+coco_weight = hf_hub_download(repo_id="akhaliq/CLIP-prefix-captioning-COCO-weights", filename="coco_weights.pt")
 import clip
 import os
 from torch import nn
@@ -234,9 +235,9 @@ def inference(img,model_name):
   model = ClipCaptionModel(prefix_length)
     
   if model_name == "COCO":
-      model_path = 'coco_weights.pt'
+      model_path = coco_weight
   else:
-      model_path = 'conceptual_weights.pt'
+      model_path = conceptual_weight
   model.load_state_dict(torch.load(model_path, map_location=CPU)) 
   model = model.eval() 
   device = CUDA(0) if is_gpu else "cpu"