Update app.py

app.py

@@ -78,12 +78,12 @@ def submit_form():
             "success": True,
             "summary": f"GradCAM++ analysis completed with magnification {data['magval']}",
             "details": "Nucleus and cytoplasm segmented successfully",
-            "classification": result_dict[result]["class1"],
+            "classification": result_dict["classification"],
             "results": {
-                "originalImage": result_dict[result]["image1"],
-                "heatmapImage": result_dict[result]["inter1"],
-                "maskImage": result_dict[result]["mask1"],
-                "tableImage": result_dict[result]["table1"]
+                "originalImage": result_dict["image1"],
+                "heatmapImage": result_dict["inter1"],
+                "maskImage": result_dict["mask1"],
+                "tableImage": result_dict["table1"]
             }
         })