Create app.py

app.py

@@ -0,0 +1,40 @@
+import gradio as gr
+from PIL import Image
+import requests
+import io
+
+def classify_image(image):
+    # Open the image file using Pillow
+    img = Image.open(io.BytesIO(image.read()))
+
+    # Convert the image to base64 string
+    buffered = io.BytesIO()
+    img.save(buffered, format="PNG")
+    img_base64 = "data:image/png;base64," + buffered.getvalue().hex()
+
+    # Send a request to the endpoint with the image data
+    response = requests.get(
+        "https://4937ea62-c0b0-414f-afbc-53be1f5b0d06.id.repl.co/scan",
+        data={"base64": img_base64}
+    )
+
+    # Process the response and return the result
+    if response.text == "sfw":
+        return "Safe for work"
+    elif response.text == "nsfw":
+        return "Not safe for work"
+    else:
+        return "Unknown"
+
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=classify_image,
+    inputs="file",
+    outputs="text",
+    title="AntiNSFW",
+    description="Classify if an image is safe for work or not.",
+    thumbnail="path_to_thumbnail_image.png"
+)
+
+# Launch the interface
+iface.launch()