Update app.py

app.py

@@ -19,10 +19,11 @@ MAFFT_PATH = "mafft/mafftdir/bin/mafft"
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # --- Paths ---
-model_dir = "models"
-boundary_path = os.path.join(model_dir, "best_boundary_aware_model.pth")
-keras_path = os.path.join(model_dir, "best_model.keras")
-kmer_path = os.path.join(model_dir, "kmer_to_index.pkl")
+from huggingface_hub import hf_hub_download
+import tempfile
+
+# Model repository and file paths
+model_repo = "GGproject10/best_boundary_aware_model"
 csv_path = "f gene clean dataset.csv"
 
 # --- Load Models ---
@@ -30,27 +31,31 @@ boundary_model = None
 keras_model = None
 kmer_to_index = None
 
-# Try to load boundary model
+# Try to load boundary model from Hugging Face Hub
 try:
+    boundary_path = hf_hub_download(repo_id=model_repo, filename="best_boundary_aware_model.pth")
     if os.path.exists(boundary_path):
         boundary_model = GenePredictor(boundary_path)
-        logging.info("Boundary model loaded successfully.")
+        logging.info("Boundary model loaded successfully from Hugging Face Hub.")
     else:
-        logging.warning(f"Boundary model file not found at {boundary_path}")
+        logging.warning(f"Boundary model file not found after download")
 except Exception as e:
-    logging.error(f"Failed to load boundary model: {e}")
+    logging.error(f"Failed to load boundary model from HF Hub: {e}")
 
-# Try to load Keras model
+# Try to load Keras model from Hugging Face Hub
 try:
+    keras_path = hf_hub_download(repo_id=model_repo, filename="best_model.keras")
+    kmer_path = hf_hub_download(repo_id=model_repo, filename="kmer_to_index.pkl")
+    
     if os.path.exists(keras_path) and os.path.exists(kmer_path):
         keras_model = load_model(keras_path)
         with open(kmer_path, "rb") as f:
             kmer_to_index = pickle.load(f)
-        logging.info("Keras model and k-mer index loaded successfully.")
+        logging.info("Keras model and k-mer index loaded successfully from Hugging Face Hub.")
     else:
-        logging.warning(f"Keras model or kmer files not found at {keras_path} or {kmer_path}")
+        logging.warning(f"Keras model or kmer files not found after download")
 except Exception as e:
-    logging.error(f"Failed to load Keras model: {e}")
+    logging.error(f"Failed to load Keras model from HF Hub: {e}")
 
 # --- Keras Prediction ---
 def predict_with_keras(sequence):