Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import spaces
 
 # Load the model and tokenizer
 model_name = "mrcuddle/SD-Prompter"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+@spaces.GPU
 # Function to generate a response
 def chat(message, history):
     # Combine the message and history into a single input