changed scheduler

app.py

@@ -8,13 +8,13 @@ from diffusers import StableDiffusionPipeline
 #function to call API here
 def show_image(prompt):
     num_images = 2
-    scheduler = PNDMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", skip_prk_steps=True)
+    scheduler = PNDMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", skip_prk_steps=True, steps_offset=1)
     model_id = "runwayml/stable-diffusion-v1-5"
 
     device = 'cuda' if  torch.cuda.is_available() else 'cpu'
     remove_safety = False
 
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16, revision="fp16", use_auth_token=True)
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", use_auth_token=True)
     if remove_safety:
         pipe.safety_checker = lambda images, clip_input: (images, False)