Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from PIL import Image
 from transformers import AutoModelForCausalLM, AutoProcessor
@@ -40,6 +41,7 @@ else:
     models['8b']['model'].eval()
     models['8b']['processor'] = models['8b']['model'].model.processor
 
+@spaces.GPU
 def convert_to_svg(image, model_choice):
     try:
         if image is None:
@@ -70,8 +72,8 @@ def convert_to_svg(image, model_choice):
         return None, None, f"Error: {str(e)}"
 
 # Create Blocks interface
-with gr.Blocks(title="Image to SVG Converter") as demo:
-    gr.Markdown("# Image to SVG Converter")
+with gr.Blocks(title="StarVector") as demo:
+    gr.Markdown("# StarVector")
     gr.Markdown("Upload an image to convert it to SVG format using StarVector model")
     
     with gr.Row():