Upload folder using huggingface_hub

README.md

@@ -1,15 +1,22 @@
 ---
-title: pashto-base-bloom
+title: pashto-base-bloom Space
 emoji: 🚀
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 5.36.2
+sdk_version: 4.36.1
 app_file: app.py
-pinned: true
+pinned: false
 license: apache-2.0
+hardware: zero-gpu-a10g
 ---
 
-# pashto-base-bloom Training Space
+# pashto-base-bloom Space
 
-Training and testing interface for pashto-base-bloom.
+This is a Space for the pashto-base-bloom model. You can:
+
+1. Test the model
+2. Train the model
+3. Fine-tune the model
+
+Uses ZeroGPU for efficient GPU acceleration.

app.py

@@ -39,8 +39,11 @@ class TrainingState:
             self.logs.pop(0)
             
     def complete_process(self):
-        elapsed = time.time() - self.start_time
-        self.add_log(f"{self.status.capitalize()} completed in {elapsed:.1f} seconds!")
+        if self.start_time is not None:
+            elapsed = time.time() - self.start_time
+            self.add_log(f"{self.status.capitalize()} completed in {elapsed:.1f} seconds!")
+        else:
+            self.add_log(f"{self.status.capitalize()} completed!")
         self.status = "idle"
         self.progress = 100
         
@@ -143,7 +146,7 @@ def get_current_status():
     
     return {
         status_box: status_text,
-        progress_bar: state.progress / 100,
+        progress_bar: state.progress,
         log_output: logs
     }
 
@@ -184,21 +187,14 @@ with gr.Blocks(title="Pashto-Base-Bloom Trainer", theme="soft") as demo:
         with gr.Row():
             with gr.Column():
                 status_box = gr.Textbox(label="Current Status", interactive=False)
-                progress_bar = gr.ProgressBar(label="Progress")
+                # Replace ProgressBar with Slider for progress indication
+                progress_bar = gr.Slider(label="Progress", minimum=0, maximum=100, value=0, interactive=False)
                 refresh_btn = gr.Button("Refresh Status", variant="secondary")
                 auto_refresh = gr.Checkbox(label="Auto-refresh every 5 seconds", value=True)
             log_output = gr.Textbox(label="Process Logs", lines=8, interactive=False)
         
-        # Auto-refresh component
-        auto_refresh_component = gr.Interval(5, interactive=False)
-        
         # Refresh actions
         refresh_btn.click(get_current_status, outputs=[status_box, progress_bar, log_output])
-        auto_refresh_component.change(
-            fn=lambda: get_current_status() if auto_refresh.value else None,
-            outputs=[status_box, progress_bar, log_output]
-        )
-        auto_refresh.change(lambda x: gr.update(interactive=x), inputs=auto_refresh, outputs=auto_refresh_component)
         
         # Initial status load
         demo.load(get_current_status, outputs=[status_box, progress_bar, log_output])

requirements.txt

@@ -1 +1,4 @@
-gradio==4.36.1
+gradio>=4.0.0
+spaces
+torch
+transformers>=4.30.0