Update app.py

app.py

@@ -21,13 +21,22 @@ loaded_tokenizer = None
 loaded_config = None
 generation_history = []
 
-def load_model_from_pickle(pickle_file):
-    """Load model from uploaded pickle file"""
+# Auto-load model on startup
+def initialize_model():
+    """Initialize model automatically on app startup"""
+    return load_model_from_pickle("best_model.pkl")
+
+def load_model_from_pickle(pickle_path="best_model.pkl"):
+    """Load model from pickle file (auto-loads on startup)"""
     global loaded_model, loaded_tokenizer, loaded_config
     
     try:
+        # Check if file exists
+        if not os.path.exists(pickle_path):
+            return f"❌ Model file not found: {pickle_path}\n\nPlease ensure best_model.pkl is uploaded to the HuggingFace Space."
+        
         # Load pickle file
-        with open(pickle_file.name, 'rb') as f:
+        with open(pickle_path, 'rb') as f:
             model_package = pickle.load(f)
         
         loaded_model = model_package['model']
@@ -61,7 +70,7 @@ def load_model_from_pickle(pickle_file):
         return config_info
         
     except Exception as e:
-        return f"❌ Error loading model: {str(e)}\n\nPlease ensure you uploaded the correct best_model.pkl file."
+        return f"❌ Error loading model: {str(e)}\n\nPlease ensure best_model.pkl is properly uploaded to the Space."
 
 def calculate_bleu_score(reference, hypothesis):
     """Calculate BLEU score between reference and generated code"""
@@ -306,25 +315,19 @@ with gr.Blocks(title="🚀 GPT-2 Pseudo-Code to Code Generator", theme=gr.themes
         with gr.Tab("💻 Code Generation"):
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("### 📥 Step 1: Load Model")
-                    model_file = gr.File(
-                        label="Upload best_model.pkl",
-                        file_types=[".pkl"],
-                        type="filepath"
-                    )
-                    load_btn = gr.Button("🔄 Load Model", variant="primary", size="lg")
+                    gr.Markdown("### � Model Status")
                     model_status = gr.Textbox(
-                        label="Model Status",
+                        label="Model Information",
                         lines=15,
                         interactive=False,
-                        placeholder="Upload model file and click 'Load Model'..."
+                        value=initialize_model()  # Auto-load on startup
                     )
             
             gr.Markdown("---")
             
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("### ✍️ Step 2: Enter Pseudo-Code")
+                    gr.Markdown("### ✍️ Enter Pseudo-Code")
                     
                     # Example selector
                     with gr.Row():
@@ -583,12 +586,6 @@ with gr.Blocks(title="🚀 GPT-2 Pseudo-Code to Code Generator", theme=gr.themes
     """)
     
     # Event handlers
-    load_btn.click(
-        fn=load_model_from_pickle,
-        inputs=[model_file],
-        outputs=[model_status]
-    )
-    
     example_dropdown.change(
         fn=show_examples,
         inputs=[example_dropdown],