Update app.py

app.py

@@ -111,12 +111,12 @@ 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 = gr.Interface(fn=image_classifier, inputs=gr.Image(type="pil"), outputs="label", title=title, description=description)
 #demo.launch(show_api=False)
 
 #interface
 
-interface = gr.Interface(
+interface1 = gr.Interface(
  fn=predict,
  inputs=[
  gr.inputs.Image(label="Input Image", type="pil"),
@@ -135,8 +135,45 @@ interface = gr.Interface(
  #examples=[[examples[i]["path"], examples[i]["label"]] for i in range(10)]
 )
 
-'''
-io2 = gr.Interface(
+title1 = "AI Generated Image Detection"
+description1 = "~ AI - ML implementation for AI image detection using older models such as VQGAN+CLIP."
+article1 = "<p style='text-align: center'>
+"""
+NOTE: To detect pictures generated using older models such as VQGAN+CLIP, please use the updated version of this detector instead.
+
+In this model i'm using a ViT model to predict whether an artistic image was generated using AI or not.
+
+The training dataset didn't include any samples generated from Midjourney 5, SDXL, or DALLE-3. But was trained on outputs of their predecessors.
+
+Scope of this tool is artistic images; that is to say, it is not a deepfake photo detector, and general computer imagery (webcams, screenshots, etc.) may throw it off.
+
+The potential indicator for this tool is to serve to detect whether an image was AI-generated or not.
+
+Images scoring as very probably artificial (e.g. 90% or higher) could be referred to a human expert for further investigation, if needed.
+Model Trained Using AutoTrain
+
+ Problem type: Binary Classification
+ Model ID: 151965872
+ CO2 Emissions (in grams): 7.9405
+
+Validation Metrics
+
+ Loss: 0.163
+ Accuracy: 0.942
+ Precision: 0.938
+ Recall: 0.978
+ AUC: 0.980
+ F1: 0.958
+
+License Notice
+
+This work is licensed under a Creative Commons Attribution-NoDerivatives 4.0 International License.
+
+You may distribute and make this model available to others as part of your own web page, app, or service so long as you provide attribution. However, use of this model within text-to-image systems to evade AI image detection would be considered a "derivative work" and as such prohibited by the license terms.
+"""
+</p>"
+
+interface1 = gr.Interface(
  fn=image_classifier,
  inputs=[
  gr.inputs.Image(label="Input Image", type="pil"),
@@ -149,10 +186,11 @@ io2 = gr.Interface(
  ],
  
  theme = gr.themes.Soft(),
- title = "AI Art Detector",
- description = description,
+ title = title1,
+ description = description1,
+ article = article1
 )
-'''
+
 gr.TabbedInterface(
- [interface, demo], ["Deepfake Image Detection", "AI ART Detection"]
+ [interface1, interface2], ["Deepfake Image Detection", "AI Image Detection"]
 ).launch() #share=True)