Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
| 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
|