Add @spaces.GPU decorator

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import spaces
 
 # Load the model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("braindeck/text2text", trust_remote_code=True, subfolder="checkpoints/model")
 model = AutoModelForCausalLM.from_pretrained("braindeck/text2text", trust_remote_code=True, torch_dtype=torch.bfloat16, device_map="auto", subfolder="checkpoints/model")
 
+@spaces.GPU
 def generate_response(prompt):
     """
     Generates a response from the model.
@@ -20,7 +22,7 @@ def generate_response(prompt):
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Text-to-Text Generation with DeepSeek-R1-Distill-Qwen-7B")
+    gr.Markdown("# Fine-tuned Text-to-Text Generation")
     gr.Markdown("Enter a prompt and the model will generate a response.")
     
     with gr.Row():
@@ -39,4 +41,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()