yeq6x commited on
Commit
4a7bff4
1 Parent(s): 6d52164
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import spaces
3
 
 
4
  def process_image(input_image, mode, weight1=None, weight2=None):
5
  print(f"Processing image with mode={mode}, weight1={weight1}, weight2={weight2}")
6
  # 既存の画像処理ロジック
 
1
  import gradio as gr
2
  import spaces
3
 
4
+ @spaces.GPU
5
  def process_image(input_image, mode, weight1=None, weight2=None):
6
  print(f"Processing image with mode={mode}, weight1={weight1}, weight2={weight2}")
7
  # 既存の画像処理ロジック