GUI cleanup

app.py

@@ -20,7 +20,7 @@ def load_model():
         tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
         model = AutoModelForCausalLM.from_pretrained(
             MODEL_NAME,
-            torch_dtype=torch.float32,  # Use float32 for CPU
+            dtype=torch.float32,  # Use float32 for CPU
             device_map="auto"
         )
         
@@ -134,20 +134,10 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column(scale=2):
-            # Model loading section
+            # Model loading section (Removed as per request)
+            # The model is loaded automatically at startup.
+
             with gr.Group():
-                gr.Markdown("### 📦 Model Management")
-                model_status = gr.Textbox(
-                    label="Model Status",
-                    value="Model is loading automatically...",
-                    interactive=False
-                )
-                load_btn = gr.Button(
-                    "🔄 Reload Model",
-                    variant="primary",
-                    size="lg"
-                )
-                
                 # Generation parameters
                 gr.Markdown("### ⚙️ Generation Parameters")
                 
@@ -236,11 +226,7 @@ with gr.Blocks() as demo:
         )
     
     # Event handlers
-    load_btn.click(
-        fn=load_model,
-        outputs=[model_status],
-        api_visibility="public"
-    )
+
     
     generate_btn.click(
         fn=generate_text,
@@ -278,7 +264,8 @@ with gr.Blocks() as demo:
     )
 
 # Auto-load the model at startup
-load_model()
+load_status = load_model()
+print(f"Startup load status: {load_status}")
 
 # Launch the application
 demo.launch(

requirements.txt

@@ -1,3 +1,4 @@
+accelerate
 torch
 transformers
-gradio
+gradio