Upload 2 files

Dockerfile

@@ -13,7 +13,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Create necessary directories (but don't download models here!)
-RUN mkdir -p /app/modelsBioembed /app/models_folder /app/Samples
+RUN mkdir -p /app/modelsBioembed /app/models_folder /app/Samples /app/numba_cache /app/hf_cache
 
 # Copy the entire project to the container
 COPY . .

app.py

@@ -21,6 +21,9 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.environ["TMPDIR"] = bio_model_dir
 os.environ["TEMP"] = bio_model_dir
 os.environ["TMP"] = bio_model_dir
+os.environ['NUMBA_CACHE_DIR'] = '/app/numba_cache'
+os.environ['TRANSFORMERS_CACHE'] = '/app/hf_cache'
+
 
 # 🔗 Dropbox Links for Model Files
 DROPBOX_LINKS = {
@@ -31,33 +34,33 @@ DROPBOX_LINKS = {
     "special_tokens_map.json": "https://www.dropbox.com/scl/fi/t3lvmp5x28d1zjac3j7ec/special_tokens_map.json?rlkey=z2xbompa54iu4y9qgb5bvmfc9&st=zrxlpjdt&dl=1"
 }
 
-# 📥 Function to Download Model Files
-def download_model_files():
-    for filename, url in DROPBOX_LINKS.items():
-        file_path = os.path.join(bio_model_dir, filename)
-        if not os.path.exists(file_path):  # Avoid re-downloading
-            print(f"Downloading {filename}...")
-            response = requests.get(url, stream=True)
-            if response.status_code == 200:
-                with open(file_path, "wb") as f:
-                    for chunk in response.iter_content(chunk_size=1024):
-                        f.write(chunk)
-                print(f"Downloaded: {filename}")
-            else:
-                print(f"Failed to download {filename}")
+# # 📥 Function to Download Model Files
 # def download_model_files():
 #     for filename, url in DROPBOX_LINKS.items():
 #         file_path = os.path.join(bio_model_dir, filename)
+#         if not os.path.exists(file_path):  # Avoid re-downloading
+#             print(f"Downloading {filename}...")
+#             response = requests.get(url, stream=True)
+#             if response.status_code == 200:
+#                 with open(file_path, "wb") as f:
+#                     for chunk in response.iter_content(chunk_size=1024):
+#                         f.write(chunk)
+#                 print(f"Downloaded: {filename}")
+#             else:
+#                 print(f"Failed to download {filename}")
+def download_model_files():
+    for filename, url in DROPBOX_LINKS.items():
+        file_path = os.path.join(bio_model_dir, filename)
 
-#         print(f"Downloading {filename} (forcing overwrite)...")
-#         response = requests.get(url, stream=True)
-#         if response.status_code == 200:
-#             with open(file_path, "wb") as f:
-#                 for chunk in response.iter_content(chunk_size=1024):
-#                     f.write(chunk)
-#             print(f"Downloaded: {filename}")
-#         else:
-#             print(f"Failed to download {filename}")
+        print(f"Downloading {filename} (forcing overwrite)...")
+        response = requests.get(url, stream=True)
+        if response.status_code == 200:
+            with open(file_path, "wb") as f:
+                for chunk in response.iter_content(chunk_size=1024):
+                    f.write(chunk)
+            print(f"Downloaded: {filename}")
+        else:
+            print(f"Failed to download {filename}")
 
 # 📥 Download models before starting
 download_model_files()