nuwandaa commited on
Commit
4d50073
·
1 Parent(s): 9545ebe

Add spaces

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ if is_attention_slicing_enabled:
86
  if is_cpu_offload_enabled:
87
  pipeline.enable_model_cpu_offload()
88
 
89
-
90
  def remove(gradio_image, rm_guidance_scale=9, num_inference_steps=50, seed=42, strength=0.8):
91
  try:
92
  generator = torch.Generator('cuda').manual_seed(seed)
 
86
  if is_cpu_offload_enabled:
87
  pipeline.enable_model_cpu_offload()
88
 
89
+ @spaces.GPU
90
  def remove(gradio_image, rm_guidance_scale=9, num_inference_steps=50, seed=42, strength=0.8):
91
  try:
92
  generator = torch.Generator('cuda').manual_seed(seed)