Update app.py

app.py

@@ -1,9 +1,3 @@
-import spaces
-
-@spaces.GPU
-def my_gpu_function():
-    # Your code here
-
 import gradio as gr
 import os
 import torch
@@ -23,6 +17,7 @@ from peft import (
     prepare_model_for_kbit_training,
     PeftModel
 )
+import spaces
 
 # Set environment variable for cache directory
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
@@ -75,6 +70,7 @@ def sample_from_csv(csv_file, sample_size=100):
     
     return Dataset.from_dict({"text": texts})
 
+@spaces.GPU
 def finetune_model(csv_file, sample_size=100, num_epochs=3, progress=gr.Progress()):
     """Fine-tune the model and return results"""
     # Check GPU