Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline, AutoImageProcessor, AutoModelForImageClassification
+from transformers import pipeline
 from PIL import Image
 
 # Title of the web app
@@ -19,12 +19,8 @@ if uploaded_file is not None:
     image = Image.open(uploaded_file)
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
-    # Load the model and processor
-    processor = AutoImageProcessor.from_pretrained("Falconsai/nsfw_image_detection")
-    model = AutoModelForImageClassification.from_pretrained("Falconsai/nsfw_image_detection")
-
-    # Use the pipeline for image classification
-    pipe = pipeline("image-classification", model=model, feature_extractor=processor)
+    # Load the model using the pipeline
+    pipe = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
 
     # Classify the image
     with st.spinner('Classifying...'):