oh my god shut up about the safety checker already

app.py

@@ -1,4 +1,9 @@
 import torch
+import numpy as np
+import gradio as gr
+from scipy import signal
+from diffusers.utils import logging
+logging.set_verbosity_error()
 from diffusers.loaders import AttnProcsLayers
 from transformers import CLIPTextModel, CLIPTokenizer
 from modules.beats.BEATs import BEATs, BEATsConfig
@@ -21,9 +26,7 @@ from diffusers import (
     KDPM2AncestralDiscreteScheduler,
     KDPM2DiscreteScheduler,
 )
-import numpy as np
-import gradio as gr
-from scipy import signal
+
 
 class AudioTokenWrapper(torch.nn.Module):
     """Simple wrapper module for Stable Diffusion that holds all the models together"""
@@ -198,7 +201,6 @@ def greet(audio, steps=25, scheduler="ddpm"):
     image = pipeline(prompt, num_inference_steps=steps, guidance_scale=8.5, generator=generator).images[0]
     return image
 
-
 lora = False
 repo_id = "philz1337/reliberate"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")