Update predictor.py

predictor.py

@@ -1,10 +1,7 @@
 # -*- coding: utf-8 -*-
-"""predictor.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1JURb-0j-R4LWK3oxeGrNxpJm3V6nnX02
+"""
+Gene Prediction Model - predictor.py
+Boundary-aware deep learning model for gene prediction
 """
 
 import torch
@@ -311,7 +308,7 @@ class EnhancedPostProcessor:
 class GenePredictor:
     """Handles gene prediction using the trained boundary-aware model."""
 
-    def __init__(self, model_path: str = 'model/best_boundary_aware_model.pth',
+    def __init__(self, model_path: str = 'best_boundary_aware_model.pth',
                  device: str = 'cuda' if torch.cuda.is_available() else 'cpu'):
         self.device = device
         self.model = BoundaryAwareGenePredictor(input_dim=14).to(device)