Upload folder using huggingface_hub

inference.py

@@ -1,3 +1,20 @@
+import os
+import torch
+from huggingface_hub import hf_hub_download
+
+# 1. Create the weights folder in the Space environment
+os.makedirs("weights", exist_ok=True)
+
+# 2. Download the heavy .pth file from your MODEL repo to the SPACE
+# This only happens once when the Space starts up.
+if not os.path.exists("weights/vqa_model.pth"):
+    print("Downloading weights from Model Hub...")
+    hf_hub_download(
+        repo_id="PRUTHVIn/vqa_project", 
+        filename="weights/vqa_model.pth", 
+        local_dir="."
+    )
+    
 from transformers import (
     Blip2Processor,
     Blip2ForConditionalGeneration,