moved gradcam inside app

app.py

@@ -1,6 +1,3 @@
-import os
-os.system("pip install gradio==4.44.1 --no-cache-dir")
-
 import gradio as gr
 import numpy as np
 from app.predictor import predict_disease, compute_ai_risk
@@ -14,9 +11,13 @@ def predict(image):
     # 2️⃣ GradCAM heatmap
     heatmap = generate_gradcam(tensor, original)
 
-    # Convert to displayable image
+    # Safety check (VERY IMPORTANT for Gradio image output)
+    heatmap = np.array(heatmap)
+
+    # Normalize if needed
     if heatmap.max() <= 1:
         heatmap = heatmap * 255
+
     heatmap = heatmap.astype(np.uint8)
 
     # 3️⃣ Risk prediction
@@ -26,14 +27,14 @@ def predict(image):
 🫀 AI Clinical Report
 
 Disease: {label}
-Confidence: {round(confidence*100,2)} %
+Confidence: {round(confidence * 100, 2)} %
 Risk Level: {risk}
 """
 
     return report, heatmap
 
 
-# 🔴 BLOCKS UI (no more Interface)
+# 🔴 UI
 with gr.Blocks(title="CardioGuard AI") as demo:
 
     gr.Markdown("# 🫀 CardioGuard AI")
@@ -53,4 +54,10 @@ with gr.Blocks(title="CardioGuard AI") as demo:
         outputs=[report_output, heatmap_output]
     )
 
-demo.launch()
+
+# ✅ IMPORTANT FIX FOR SPACES / CLOUD
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=7860,
+    share=False
+)