Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-from transformers import AutoBackbone, AutoModelForImageClassification, pipeline, AutoImageProcessor, SwinForImageClassification
+from transformers import AutoBackbone, AutoModelForImageClassification, pipeline, AutoImageProcessor, Swinv2ForImageClassification
 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 = AutoImageProcessor.from_pretrained("haywoodsloan/ai-image-detector-deploy")
-model = AutoBackbone.from_pretrained("haywoodsloan/ai-image-detector-deploy", out_indices=(1,))
+image_processor = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
+model = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy", out_indices=(1,))
 
 clf = pipeline(model=model, task="image-classification", image_processor=image_processor)