Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 import paddlehub as hub
 import urllib
 import cv2
-from datasets import load_dataset
 import re
 import os
 import requests
@@ -73,16 +72,12 @@ word_list = word_list_dataset["train"]['text']
 
 def get_result(text_prompts, style_indx):
     style = style_list[style_indx]
-    prompt = "," + style
-    for filter in word_list:
-        if re.search(rf"\b{filter}\b", text_prompts):
-            raise gr.Error("Unsafe content found. Please try again with different prompts.")
+    prompt = text_prompts + "," + style
         
     model_id = "runwayml/stable-diffusion-v1-5"
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
     pipe = pipe.to("cuda")
     
-    prompt = text_prompts
     image_output = pipe(prompt).images[0]
     
     print("file name: " + image_output.filename)
@@ -109,6 +104,7 @@ def get_music(prompt):
   with open("output.wav", "wb") as f:
     f.write(wav[0].getbuffer())
   return spec, 'output.wav'
+    
   
 def merge_video(music, img_list):
   #Convert to mp3