Upload folder using huggingface_hub
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ for n in range(1, 10):
|
|
| 16 |
|
| 17 |
@spaces.GPU
|
| 18 |
def infer(init_image, prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
|
| 19 |
-
depth = model_depth.image2depth([init_image])
|
| 20 |
return model_image.text2image(
|
| 21 |
[init_image],
|
| 22 |
depth_maps=depth,
|
|
|
|
| 16 |
|
| 17 |
@spaces.GPU
|
| 18 |
def infer(init_image, prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
|
| 19 |
+
depth = model_depth.image2depth([init_image], return_tensor=True)
|
| 20 |
return model_image.text2image(
|
| 21 |
[init_image],
|
| 22 |
depth_maps=depth,
|