Update app.py

app.py

@@ -8,6 +8,8 @@ import onnxruntime as rt
 import pandas as pd
 from PIL import Image
 
+import spaces
+
 TITLE = "WaifuDiffusion Tagger"
 DESCRIPTION = """
 Demo for the WaifuDiffusion tagger models
@@ -129,7 +131,7 @@ class Predictor:
         self.general_indexes = sep_tags[2]
         self.character_indexes = sep_tags[3]
 
-        model = rt.InferenceSession(model_path)
+        model = rt.InferenceSession(model_path, providers=["CUDAExecutionProvider"], provider_options=[{"device_id": 0}])
         _, height, width, _ = model.get_inputs()[0].shape
         self.model_target_size = height
 
@@ -167,6 +169,7 @@ class Predictor:
 
         return np.expand_dims(image_array, axis=0)
 
+    @spaces.GPU
     def predict(
         self,
         image,