Update app.py

app.py

@@ -181,6 +181,7 @@ def predict_image(img, confidence_threshold):
         response5 = response5_raw
         print(response5)
         label_5 = f"Result: {response5}"
+        result_5output = [5, 0.0, 0.0, 'MAINTENANCE']
     except Exception as e:
         label_5 = f"Error: {str(e)}"
     
@@ -192,7 +193,8 @@ def predict_image(img, confidence_threshold):
         "Swin/SDXL-FLUX": label_4,
         "GOAT": label_5
     }
-    return img_pil, combined_results
+    combined_outputs = { result_1output, result_2output, result_3output, result_4output, result_5output}
+    return img_pil, combined_outputs
 
 # Define a function to generate the HTML content
 def generate_results_html(results):
@@ -201,23 +203,23 @@ def generate_results_html(results):
     <div class="container">
         <div class="row mt-4">
             <div class="col">
-                <h5>SwinV2/detect</h5>
+                <h5>SwinV2/detect <span class="badge badge-secondary">M1</span></h5>
                 <p>{results.get("SwinV2/detect", "N/A")}</p>
             </div>
             <div class="col">
-                <h5>ViT/AI-vs-Real</h5>
+                <h5>ViT/AI-vs-Real <span class="badge badge-secondary">M2</span></h5>
                 <p>{results.get("ViT/AI-vs-Real", "N/A")}</p>
             </div>
             <div class="col">
-                <h5>Swin/SDXL</h5>
+                <h5>Swin/SDXL <span class="badge badge-secondary">M3</span></h5>
                 <p>{results.get("Swin/SDXL", "N/A")}</p>
             </div>
             <div class="col">
-                <h5>Swin/SDXL-FLUX</h5>
+                <h5>Swin/SDXL-FLUX <span class="badge badge-secondary">M4</span></h5>
                 <p>{results.get("Swin/SDXL-FLUX", "N/A")}</p>
             </div>
             <div class="col">
-                <h5>GOAT</h5>
+                <h5>GOAT <span class="badge badge-secondary">M5</span></h5>
                 <p>{results.get("GOAT", "N/A")}</p>
             </div>
         </div>