Update app.py

app.py

@@ -69,7 +69,22 @@ def classify_image(image):
 
     except Exception as e:
         # Log and return detailed error messages
-        print(f"Error during classification:
+        print(f"Error during classification: {e}")
+        return {"Error": str(e)}
+
+# Step 3: Set Up Gradio Interface
+iface = gr.Interface(
+    fn=classify_image,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(label="Classification Results"),  # Display probabilities as progress bars
+    title="Content Safety Classification",
+    description="Upload an image to classify it as 'safe' or 'unsafe' with corresponding probabilities.",
+)
+
+if __name__ == "__main__":
+    print("Launching the Gradio interface...")
+    iface.launch()
+