Update app.py

app.py

@@ -85,7 +85,7 @@ def process_files(ai_files, real_files):
 
 def evaluate_model(temp_dir):
     labels, preds, images = [], [], []
-    false_positives, false_negatives = [], []
+    false_positives, false_negatives = []
     detector = AIDetector()
     
     total_images = sum(len(files) for _, _, files in os.walk(temp_dir))
@@ -232,7 +232,6 @@ def generate_pdf(accuracy, roc_score, report_html, confusion_matrix_plot):
     buffer.seek(0)
     return buffer
 
-
 def load_url(url):
     try:
         urllib.request.urlretrieve(url, "temp_image.png")
@@ -261,9 +260,10 @@ def create_gradio_interface():
 
                 with gr.Group():
                     with gr.Box():
-                         gr.HTML(
-                            """<b>Model: Newhouse AI Image Detection Model v1.1</b><br>
-                            <i>This model has been trained on flux dev, flux schnell, Stable Diffusion 1.2, SDXL, and 3.5. The changes from 1.0 are that the model has had additional fine tuning on solid background images</i>"""
+                        # Add your note here about training:
+                        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")
@@ -292,7 +292,11 @@ def create_gradio_interface():
                     individual_process_btn = gr.Button("Process Individual Files", interactive=False)
 
                 with gr.Group():
-                    gr.Markdown(f"### Newhouse AI Image Detection Model v1.0")
+                    # Add the same or similar note here if you want it repeated:
+                    gr.Markdown(
+                        """### Newhouse AI Image Detection Model v1.0  
+                        **This model has been trained on flux dev, flux schnell, Stable Diffusion 1.2, SDXL, and 3.5.**"""
+                    )
                     output_acc = gr.Label(label="Accuracy")
                     output_roc = gr.Label(label="ROC Score")
                     output_report = gr.HTML(label="Classification Report")
@@ -371,4 +375,4 @@ if __name__ == "__main__":
         show_api=False,
         max_threads=24,
         show_error=True
-    )
+    )