Update app.py

app.py

@@ -1,12 +1,25 @@
 import gradio as gr
 import torch
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification, pipeline
 
 models=[
     "Nahrawy/AIorNot",
     "RishiDarkDevil/ai-image-det-resnet152",
     "arnolfokam/ai-generated-image-detector",
 ]
+pipe = pipeline("image-classification", "umm-maybe/AI-image-detector")
+
+def image_classifier(image):
+    outputs = pipe(image)
+    results = {}
+    for result in outputs:
+        results[result['label']] = result['score']
+    return results
+
+
+
+#demo = gr.Interface(fn=image_classifier, inputs=gr.Image(type="pil"), outputs="label", title=title, description=description)
+#demo.launch(show_api=False)