Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
-from transformers import AutoBackbone, AutoModelForImageClassification, pipeline, AutoImageProcessor, Swinv2ForImageClassification
+# from transformers import AutoBackbone, AutoModelForImageClassification, AutoImageProcessor, Swinv2ForImageClassification
+from transformers import pipeline, AutoImageProcessor, Swinv2ForImageClassification, Swinv2Model
 from torchvision import transforms
 
 # model = AutoModelForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
 # image_processor = AutoImageProcessor.from_pretrained("haywoodsloan/ai-image-detector-deploy")
-
-image_processor = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
-model = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy", out_indices=(1,))
+image_processor = AutoImageProcessor.from_pretrained("haywoodsloan/ai-image-detector-deploy")
+# image_processor = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
+model = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
 
 clf = pipeline(model=model, task="image-classification", image_processor=image_processor)