[Fix] return gradio plot for heatmap

app.py

@@ -30,7 +30,7 @@ def create_heatmap(selected_models, selected_dataset):
     )
     
     print("Heatmap generated successfully")  # Debugging
-    return fig
+    return gr.Plot(fig)
 
 def validate_inputs(selected_models, selected_dataset):
     if not selected_models: