🚀 Add ZeroGPU support for better performance

app.py

@@ -5,6 +5,7 @@ import random
 from datetime import datetime
 from datasets import load_dataset
 import pandas as pd
+import spaces
 
 # Global state
 class TrainingState:
@@ -57,6 +58,7 @@ class TrainingState:
         if len(self.logs) > 15:
             self.logs.pop(0)
 
+@spaces.GPU
     def complete_process(self):
         elapsed = time.time() - self.start_time
         self.add_log(f"🏁 {self.status.capitalize()} completed in {elapsed:.1f}s")
@@ -86,6 +88,7 @@ def test_model(text):
     ]
     return random.choice(options)
 
+@spaces.GPU
 def simulate_process(duration, process_type, data_size):
     if process_type == "train":
         state.start_training(data_size)