fffiloni commited on
Commit
7463531
1 Parent(s): 2f9bd03

ready for ZeroGPU

Browse files
Files changed (1) hide show
  1. demo_gradio.py +2 -1
demo_gradio.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import huggingface_hub
2
 
3
  huggingface_hub.snapshot_download(
@@ -89,7 +90,7 @@ transform = Compose(
89
 
90
  model.eval()
91
 
92
-
93
  def greet(input_image, material_exemplar):
94
 
95
  """
 
1
+ import spaces
2
  import huggingface_hub
3
 
4
  huggingface_hub.snapshot_download(
 
90
 
91
  model.eval()
92
 
93
+ @spaces.GPU()
94
  def greet(input_image, material_exemplar):
95
 
96
  """