Update app.py
Browse files
app.py
CHANGED
@@ -44,6 +44,7 @@ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torc
|
|
44 |
pipe = pipe.to("cuda")
|
45 |
|
46 |
def diffuse(source_img):
|
|
|
47 |
prompt = "space crew in a spaceship by Claude Monet"
|
48 |
image = pipe(prompt, source_img).images[0]
|
49 |
print(f"DIFFUSED IMG: {image}")
|
|
|
44 |
pipe = pipe.to("cuda")
|
45 |
|
46 |
def diffuse(source_img):
|
47 |
+
source_img = Image.open(source_img)
|
48 |
prompt = "space crew in a spaceship by Claude Monet"
|
49 |
image = pipe(prompt, source_img).images[0]
|
50 |
print(f"DIFFUSED IMG: {image}")
|