Update main.py

main.py

@@ -18,6 +18,19 @@ mongoserver = os.getenv("MONGO_URI")
 #import models.downloadModels
 #models.downloadModels.download_all_models()
 
+import shutil
+
+# Ensure the ./models directory exists
+os.makedirs("models", exist_ok=True)
+
+# Move/copy each model to the expected path
+shutil.copy(model_paths["layer1cnn_aanan.pth"], "models/layer1cnn_aanan.pth")
+shutil.copy(model_paths["layer2bio_00"], "models/layer2bio_00")
+shutil.copy(model_paths["layer2bio_01"], "models/layer2bio_01")
+shutil.copy(model_paths["layer2non_00"], "models/layer2non_00")
+shutil.copy(model_paths["layer2non_01"], "models/layer2non_01")
+shutil.copy(model_paths["layer3_cnn.keras"], "models/layer3_cnn.keras")
+
 from huggingface_hub import hf_hub_download
 from huggingface_hub import login
 login(token=os.getenv("HF_TOKEN"))