Update app.py

app.py

@@ -1,39 +1,19 @@
 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(
+def anti_nsfw(image):
+    resp = requests.get(
         "https://4937ea62-c0b0-414f-afbc-53be1f5b0d06.id.repl.co/scan",
-        data={"base64": img_base64}
+        data={"base64": image}
     )
+    return resp.text
 
-    # 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",
+    fn=anti_nsfw,
+    inputs="image",
     outputs="text",
     title="AntiNSFW",
-    description="Classify if an image is safe for work or not.",
+    description="Check if an image is safe for work (SFW) or not safe for work (NSFW)."
 )
 
-# Launch the interface
 iface.launch()