Update app.py

app.py

@@ -5,8 +5,13 @@ import requests
 from io import BytesIO
 
 # Load the model and processor
-processor = ViTImageProcessor.from_pretrained('AdamCodd/vit-base-nsfw-detector')
-model = AutoModelForImageClassification.from_pretrained('AdamCodd/vit-base-nsfw-detector')
+@st.experimental_singleton
+def load_model():
+    processor = ViTImageProcessor.from_pretrained('AdamCodd/vit-base-nsfw-detector')
+    model = AutoModelForImageClassification.from_pretrained('AdamCodd/vit-base-nsfw-detector')
+    return processor, model
+
+processor, model = load_model()
 
 # Define prediction function
 def predict_image(image):
@@ -27,8 +32,9 @@ def predict_image(image):
 # Streamlit app
 st.title("NSFW Image Classifier")
 
-# URL input
-image_url = st.text_input("Enter Image URL", placeholder="Enter image URL here")
+# Get image URL from query parameters
+image_url = st.experimental_get_query_params().get('image_url', [None])[0]
+
 if image_url:
     try:
         # Load image from URL
@@ -43,3 +49,5 @@ if image_url:
         st.write(f"Predicted Class: {prediction}")
     except Exception as e:
         st.write(f"Error: {e}")
+else:
+    st.write("Please provide an image URL using the 'image_url' query parameter.")