ttoosi commited on
Commit
a83cee7
·
verified ·
1 Parent(s): 2d0241c

Update app.py

Browse files

decoration @spaces.GPU

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ os.makedirs("stimuli", exist_ok=True)
18
  # Initialize model
19
  model = GenerativeInferenceModel()
20
 
 
21
  def run_inference(image, model_type, illusion_type, eps_value, num_iterations):
22
  # Convert eps to float
23
  eps = float(eps_value)
 
18
  # Initialize model
19
  model = GenerativeInferenceModel()
20
 
21
+ @spaces.GPU
22
  def run_inference(image, model_type, illusion_type, eps_value, num_iterations):
23
  # Convert eps to float
24
  eps = float(eps_value)