disable safety checker (#1)

- disable safety checker (aa6e3c54fd97aff96c17a0573f6ac22f56b078f0)


Co-authored-by: oat <[email protected]>

app.py

@@ -19,6 +19,8 @@ pipe2 = pipe2.to(device)
 
 spectro_from_wav = gr.Interface.load("spaces/fffiloni/audio-to-spectrogram")
 
+def dummy_checker(images, **kwargs): return images, False
+
 def predict(prompt, negative_prompt, audio_input, duration):
     # if audio_input == None :
     return classic(prompt, negative_prompt, duration)
@@ -26,6 +28,7 @@ def predict(prompt, negative_prompt, audio_input, duration):
     # return style_transfer(prompt, negative_prompt, audio_input)
 
 def classic(prompt, negative_prompt, duration):
+    pipe.safety_checker = dummy_checker
     spec = pipe(prompt, negative_prompt=negative_prompt, height=512, width=512).images[0]
     print(spec)
     wav = wav_bytes_from_spectrogram_image(spec)