Update app.py

app.py

@@ -76,7 +76,7 @@ def get_result(text_prompts, style_indx):
     prompt = style + "," + text_prompts
         
     
-    image_output = pipe(prompt).images[0]
+    image_output = pipe(prompt).images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
     
     print("Generated image with prompt " + prompt)
     
@@ -96,7 +96,7 @@ def get_result(text_prompts, style_indx):
     return {spec_result:spec_image, video_result:video_merged, status_text:'Success'}
 
 def get_music(prompt):
-  spec = pipe(prompt).images[0]
+  spec = pipe2(prompt).images[0]
   print(spec)
   wav = wav_bytes_from_spectrogram_image(spec)
   with open("output.wav", "wb") as f: