Update app.py

app.py

@@ -27,14 +27,19 @@ def predict_image(image):
 # Streamlit app
 st.title("NSFW Image Classifier")
 
-# Upload image file
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption='Uploaded Image.', use_column_width=True)
-    st.write("")
-    st.write("Classifying...")
+# URL input
+image_url = st.text_input("Enter Image URL", placeholder="Enter image URL here")
+if image_url:
+    try:
+        # Load image from URL
+        response = requests.get(image_url)
+        image = Image.open(BytesIO(response.content))
+        st.image(image, caption='Image from URL', use_column_width=True)
+        st.write("")
+        st.write("Classifying...")
 
-    # Predict and display result
-    prediction = predict_image(image)
-    st.write(f"Predicted Class: {prediction}")
+        # Predict and display result
+        prediction = predict_image(image)
+        st.write(f"Predicted Class: {prediction}")
+    except Exception as e:
+        st.write(f"Error: {e}")