Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
-from transformers import AutoModelForImageClassification, pipeline, AutoImageProcessor, SwinForImageClassification
+from transformers import AutoBackbone, AutoModelForImageClassification, pipeline, AutoImageProcessor, SwinForImageClassification
 from torchvision import transforms
 
-model = AutoModelForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
+# 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,))
+
 clf = pipeline(model=model, task="image-classification", image_processor=image_processor)
 
 class_names = ['artificial', 'real']