Update app.py (#1)

- Update app.py (5f847d391a5ff8708a5d26f214d0e82d7e962afa)


Co-authored-by: Ci Splunk <[email protected]>

app.py

@@ -1,5 +1,5 @@
 from os.path import basename, splitext
-
+import spaces
 import gradio as gr
 from huggingface_hub import hf_hub_download
 
@@ -8,7 +8,7 @@ from onnx_inference import vectorize_image
 
 MODEL_PATH = hf_hub_download("nopperl/marked-lineart-vectorizer", "model.onnx")
 
-
+@spaces.GPU
 def predict(input_image_path, threshold, stroke_width):
     output_filepath = splitext(basename(input_image_path))[0] + ".svg"
     for recons_img in vectorize_image(input_image_path, model=MODEL_PATH, output=output_filepath, threshold_ratio=threshold, stroke_width=stroke_width):