Update app.py

app.py

@@ -17,7 +17,7 @@ def load_sentiment_analyzer():
 # Load text-to-image model
 @st.cache_resource
 def load_text2img():
-    model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch_dtype)
+    model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch_dtype, safety_checker = None, requires_safety_checker = False)
     model = model.to(device)
     return model
 
@@ -69,7 +69,7 @@ def generate_prompt(emotions):
     return "Create a multi-style artwork combining " + ", ".join(prompt_parts)
 
 def generate_image(prompt):
-    image = text2img(prompt, num_inference_steps=10).images[0]
+    image = text2img(prompt, num_inference_steps=5).images[0]
     return image
 
 def main():