Update app.py

app.py

@@ -61,10 +61,12 @@ if is_colab:
   pipe = StableDiffusionPipeline.from_pretrained(
       current_model.path,
       torch_dtype=torch.float16,
-      scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
-      safety_checker=lambda images, clip_input: (images, False)
+      revision="fp16" if torch.cuda.is_available() else "fp32",
+      torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+      scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
       )
 
+
 else:
   pipe = StableDiffusionPipeline.from_pretrained(
       model_id,