Update app.py

app.py

@@ -11,6 +11,20 @@ from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
 from pytorch_grad_cam.utils.image import show_cam_on_image
 
+#ai-pict-detect
+from transformers import pipeline
+
+pipe = pipeline("image-classification", "nightfury/AI-picture-detector")
+
+def image_classifier(image):
+    outputs = pipe(image)
+    results = {}
+    for result in outputs:
+        results[result['label']] = result['score']
+    return results
+#ai-pict-detect
+
+
 #gr.themes.Soft()
 #gr.themes.builder()
     
@@ -97,6 +111,9 @@ title = "Deepfake Image Detection"
 description = "~ AI - ML implementation for fake and real image detection..."
 article = "<p style='text-align: center'>...</p>"
 
+demo = gr.Interface(fn=image_classifier, inputs=gr.Image(type="pil"), outputs="label", title=title, description=description)
+demo.launch(show_api=False)
+
 interface = gr.Interface(
     fn=predict,
     inputs=[
@@ -109,6 +126,7 @@ interface = gr.Interface(
         gr.outputs.Image(label="Face with Explainability", type="pil")
         #ValueError: Invalid value for parameter `type`: auto. Please choose from one of: ['numpy', 'pil', 'filepath']
     ],
+    
     theme = gr.themes.Soft(),
     title = title,
     description = description,