Update app.py

app.py

@@ -57,6 +57,7 @@ def run_training(status_box, time_box, download_file, model_size_box, log_box):
     size_mb = round(os.path.getsize(ZIP_FILE) / (1024 * 1024), 2)
     model_size_box.value = f"📦 Model Size: {size_mb} MB"
 
+    download_file.value = ZIP_FILE
     download_file.visible = True
     status_box.value = "✅ Done! You can now download your trained model."
 
@@ -80,7 +81,7 @@ with gr.Blocks() as demo:
         model_size_box = gr.Textbox(label="Final Model Size", interactive=False)
 
     log_box = gr.Textbox(label="Live Training Logs", lines=20, interactive=False)
-    download_file = gr.File(label="📥 Download Trained Model (.zip)", value=ZIP_FILE, visible=False)
+    download_file = gr.File(label="📥 Download Trained Model (.zip)", visible=False)
 
     train_btn.click(fn=start_training, inputs=[status_box, time_box, download_file, model_size_box, log_box], outputs=[status_box])