Update app.py

app.py

@@ -17,7 +17,7 @@ from reportlab.pdfgen import canvas
 from bs4 import BeautifulSoup
 
 
-MODEL_NAME = "cmckinle/sdxl-flux-detector"
+MODEL_NAME = "cmckinle/sdxl-flux-detector_v1.1"
 LABELS = ["AI", "Real"]
 
 class AIDetector:
@@ -261,7 +261,10 @@ def create_gradio_interface():
 
                 with gr.Group():
                     with gr.Box():
-                        gr.HTML(f"""<b>Model: Newhouse AI Image Detection Model v1.0</b>""")
+                         gr.HTML(
+                            """<b>Model: Newhouse AI Image Detection Model v1.0</b><br>
+                            <i>This model has been trained on flux dev, flux schnell, Stable Diffusion 1.2, SDXL, and 3.5.</i>"""
+                        )
                         output_html = gr.HTML()
                         output_label = gr.Label(label="Output")