Update main.py

main.py

@@ -7,13 +7,33 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.vector_stores.faiss import FaissVectorStore
 import faiss
 import os
-from huggingface_hub import login
+from huggingface_hub import login, snapshot_download
+from huggingface_hub.utils import configure_http_backend
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
+import requests
 
 app = FastAPI()
 
 # Set HF_HOME to a writable directory
 os.environ["HF_HOME"] = "/app/.cache"
 
+# Configure requests with retries and a longer timeout
+def custom_http_backend():
+    session = requests.Session()
+    retries = Retry(
+        total=3,  # Retry 3 times
+        backoff_factor=1,  # Wait 1, 2, 4 seconds between retries
+        status_forcelist=[429, 500, 502, 503, 504],  # Retry on these HTTP status codes
+    )
+    adapter = HTTPAdapter(max_retries=retries)
+    session.mount("https://", adapter)
+    session.timeout = 60  # Increase timeout to 60 seconds
+    return session
+
+# Set the custom HTTP backend for huggingface_hub
+configure_http_backend(backend_factory=custom_http_backend)
+
 # Log in to Hugging Face
 hf_token = os.getenv("HF_TOKEN")
 if not hf_token: