Upload app.py

app.py

@@ -1,5 +1,5 @@
-import spaces
 import gradio as gr
+import spaces
 import onnxruntime as ort
 import numpy as np
 from PIL import Image, ImageDraw, ImageFont
@@ -374,6 +374,7 @@ def initialize_model():
         labels_data, _, tag_to_category_map = load_tag_mapping(tag_mapping_path)
         print("Model and labels loaded.")
 
+@spaces.GPU()
 def predict(image_input, gen_threshold, char_threshold, output_mode):
     """Gradioインターフェース用の予測関数"""
     initialize_model() # モデルがロードされていなければロード