Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ pipeline = StableDiffusionInpaintPipeline.from_pretrained(
|
|
| 7 |
torch_dtype=torch.float16,
|
| 8 |
)
|
| 9 |
|
| 10 |
-
pipeline = pipeline.to("cuda")
|
| 11 |
|
| 12 |
def predict(mask_img):
|
| 13 |
prompt = "a green frog, highly detailed, natural lighting"
|
|
|
|
| 7 |
torch_dtype=torch.float16,
|
| 8 |
)
|
| 9 |
|
| 10 |
+
#pipeline = pipeline.to("cuda")
|
| 11 |
|
| 12 |
def predict(mask_img):
|
| 13 |
prompt = "a green frog, highly detailed, natural lighting"
|