Load model with pipeline rather than use inference API

app.py

@@ -1,4 +1,14 @@
 import gradio as gr
-iface = gr.Interface.load("models/umm-maybe/AI-image-detector")
+from transformers import pipeline
 
-iface.launch()
+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")
+demo.launch()