h-siyuan commited on
Commit
5cbd5b0
·
verified ·
1 Parent(s): c8f7f4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -50,6 +50,7 @@ def array_to_image_path(image_array):
50
  img.save(filename)
51
  return os.path.abspath(filename)
52
 
 
53
  def run_showui(image, query):
54
  """Main function for inference."""
55
  image_path = array_to_image_path(image)
 
50
  img.save(filename)
51
  return os.path.abspath(filename)
52
 
53
+ @spaces.GPU
54
  def run_showui(image, query):
55
  """Main function for inference."""
56
  image_path = array_to_image_path(image)