Update app.py

app.py

@@ -1,12 +1,17 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
 # Load model
 tokenizer = AutoTokenizer.from_pretrained("vinoku89/svg-code-generator")
 model = AutoModelForCausalLM.from_pretrained("vinoku89/svg-code-generator")
 
+# 💡 Move model to GPU if available
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model.to(device)
+
 def generate_svg(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt")
+    inputs = tokenizer(prompt, return_tensors="pt").to(device)
     
     outputs = model.generate(
         **inputs,