Update app.py

app.py

@@ -45,12 +45,12 @@ def generate_sketch(prompt, num_inference_steps, guidance_scale):
     # Check the classification results
     for result in image_classification:
         if result['label'] == 'nsfw' and result['score'] > NSFW_THRESHOLD: 
-           return "Inappropriate content detected. Please try another prompt." #return gr.update(visible=False),gr.Text(value="Inappropriate content detected. Please try another prompt.")
+           return None,"Inappropriate content detected. Please try another prompt." #return gr.update(visible=False),gr.Text(value="Inappropriate content detected. Please try another prompt.")
     
     image_path = "generated_sketch.png"
     
     image.save(image_path)
-    return image_path #gr.Image(value=image_path), gr.update(visible=False)
+    return image_path,None #gr.Image(value=image_path), gr.update(visible=False)
 
 # Gradio interface with sliders for num_inference_steps and guidance_scale
 interface = gr.Interface(
@@ -61,8 +61,8 @@ interface = gr.Interface(
         gr.Slider(1.0, 10.0, value=3.5, step=0.1, label="Guidance Scale")  # Slider for guidance_scale
     ],
     outputs=[
-        gr.Image(type="file", label="Generated Sketch"),  # For image output
-        gr.Textbox(label="Message")  # For message output (in case of NSFW content)
+        gr.Image(label="Generated Sketch"), 
+        gr.Textbox(label="Message")  
     ],
     title="Kids Sketch Generator",
     description="Enter a text prompt and generate a fun sketch for kids with customizable inference steps and guidance scale."