Update app.py

app.py

@@ -1,15 +1,17 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import spaces
 
 # Load model
 tokenizer = AutoTokenizer.from_pretrained("vinoku89/svg-code-generator")
 model = AutoModelForCausalLM.from_pretrained("vinoku89/svg-code-generator")
 
-# 💡 Move model to GPU if available
+# Move model to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
+@spaces.GPU
 def generate_svg(prompt):
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
     
@@ -39,4 +41,4 @@ gradio_app = gr.Interface(
 )
 
 if __name__ == "__main__":
-    gradio_app.launch()
+    gradio_app.launch()