init

app.py

@@ -20,7 +20,7 @@ class LoggingCallback(TrainerCallback):
             error_rate = 1 - state.best_metric  # Assuming best_metric is accuracy
             print(f"Current Error Rate: {error_rate:.4f}")
 
-@spaces.GPU(duration=1800)
+@spaces.GPU(duration=120)
 def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
     try:        
         login(api_key.strip())