listen2you003 commited on
Commit
4a9ffc4
·
1 Parent(s): 50550c9

add spaces

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import json
4
  import itertools
5
  import math
6
  import os
 
7
  import time
8
  from pathlib import Path
9
 
@@ -397,6 +398,7 @@ def prepare_infer_func():
397
 
398
  return image_edit.generate_image
399
 
 
400
  def inference(prompt, ref_images, seed, size_level, infer_func=None):
401
  start_time = time.time()
402
 
 
4
  import itertools
5
  import math
6
  import os
7
+ import spaces
8
  import time
9
  from pathlib import Path
10
 
 
398
 
399
  return image_edit.generate_image
400
 
401
+ @spaces.GPU
402
  def inference(prompt, ref_images, seed, size_level, infer_func=None):
403
  start_time = time.time()
404