multimodalart HF Staff commited on
Commit
de76e75
·
verified ·
1 Parent(s): 7876c28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -162,7 +162,7 @@ model_management.load_models_gpu([
162
  ])
163
 
164
  @spaces.GPU
165
- def generate_image(prompt, structure_image, style_image, depth_strength=15, style_strength=0.5, progress=gr.Progress(track_tqdm=True)) -> str:
166
  """Main generation function that processes inputs and returns the path to the generated image."""
167
  with torch.inference_mode():
168
  # Set up CLIP
 
162
  ])
163
 
164
  @spaces.GPU
165
+ def generate_image(prompt, structure_image, style_image, depth_strength=15, style_strength=0.5) -> str:
166
  """Main generation function that processes inputs and returns the path to the generated image."""
167
  with torch.inference_mode():
168
  # Set up CLIP