Update app.py

app.py

@@ -8,6 +8,14 @@ import urllib.request
 import uuid
 uid=uuid.uuid4()
 
+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
 
 
 def softmax(vector):