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
|