Spaces:

amazonaws-sp
/

zm

Sleeping

amazonaws-sp commited on Apr 7, 2024

Commit

714f0db

verified ·

1 Parent(s): 6f3dd86

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,7 +63,7 @@ def generate(
         if use_img2img:
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
-            response = requests.get("https://s3.amazonaws.com/appforest_uf/f1712524718776x747602231923886500/image.png")
             init_image = Image.open(BytesIO(response.content)).convert("RGB")
             init_image = init_image.resize((768, 512))

         if use_img2img:
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
+            response = requests.get(url)
             init_image = Image.open(BytesIO(response.content)).convert("RGB")
             init_image = init_image.resize((768, 512))