changed torch.float16 to float32

app.py

@@ -17,7 +17,7 @@ This is an unofficial demo for [https://github.com/NVlabs/stylegan3](https://git
 
 model = Model()
 model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) #Change to torch.float16 for faster inference and using cpu
 #pipe = pipe.to("cuda")  # Remove or comment out this line if using CPU
 
 sg = None
@@ -87,9 +87,6 @@ with gr.Blocks(css='style.css') as image_gen_block:
         return sg.generate_story(prompt)
 
     def generate_city():
-        model_id = "runwayml/stable-diffusion-v1-5"
-        pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-        pipe = pipe.to("cuda")
         city_prompt = f"render 2048K ultra realistic render high definition a very technologically advanced city, cyberpunk, hyper realistic, detailed, ray tracing, POV-Ray, metallic, Sci-Fi"
         city_image = pipe(city_prompt).images[0]
         return city_image