Update app.py

app.py

@@ -4,8 +4,8 @@ from PIL import Image
 import requests
 
 # Load the model and processor
-processor = ViTImageProcessor.from_pretrained('yeftakun/vit-base-nsfw-detector')
-model = AutoModelForImageClassification.from_pretrained('yeftakun/vit-base-nsfw-detector')
+processor = ViTImageProcessor.from_pretrained('AdamCodd/vit-base-nsfw-detector')
+model = AutoModelForImageClassification.from_pretrained('AdamCodd/vit-base-nsfw-detector')
 
 # Define prediction function
 def predict_image(image_url):
@@ -29,9 +29,9 @@ def predict_image(image_url):
 # Create Gradio interface
 iface = gr.Interface(
     fn=predict_image,
-    inputs=gr.inputs.Textbox(label="Image URL"),
-    outputs=gr.outputs.Textbox(label="Predicted Class"),
-    title="NSFW Image Detection"
+    inputs=gr.Textbox(label="Image URL"),
+    outputs=gr.Textbox(label="Predicted Class"),
+    title="NSFW Image Classifier"
 )
 
 # Launch the interface