Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 # from transformers import AutoBackbone, AutoModelForImageClassification, AutoImageProcessor, Swinv2ForImageClassification
-from transformers import pipeline, AutoImageProcessor, Swinv2ForImageClassification, Swinv2Model
+from transformers import pipeline, AutoImageProcessor, Swinv2ForImageClassification
 from torchvision import transforms
 
 # model = AutoModelForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
@@ -17,6 +17,8 @@ def predict_image(img):
   img = transforms.ToPILImage()(img)
   img = transforms.Resize((256,256))(img)
   prediction=clf.predict(img)
+  print(clf(image))
+  print(model.config.id2label[predicted_label])
   return {class_names[i]: float(prediction[i]["score"]) for i in range(2)}
 
 image = gr.Image(label="Image to Analyze", sources=['upload'])