Upload app.py

app.py

@@ -18,7 +18,7 @@ USER_TOKEN=os.environ.get('HF_TOKEN_SD')
 #1.5
 #model_id = "runwayml/stable-diffusion-v1-5"
 #pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token = USER_TOKEN)
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=USER_TOKEN)
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype = "auto", use_auth_token = USER_TOKEN)
 
 pipe.to(device)
 
@@ -37,7 +37,7 @@ def infer(source_img):
     prompt = "abandoned buildings, empty streetscapes, surrounded by lush green vegetation, ground-level view, rusted steel, derelict, urban exploration, ruin, deserted, broken windows, burnt out, graffitti, ramshackle, homeless shelter, decay, 8k, photorealistic, hyper detailed"
     guide = 10
     steps = 40
-    diffstr = 0.7
+    diffstr = 0.3
     seed = random.randint(0,2147483647)
 
     generator = torch.Generator("cpu").manual_seed(seed)