Update app.py

app.py

@@ -86,11 +86,18 @@ def predict_count(input_image):
     predicted_count = int(output.detach().cpu().sum().numpy())
     density_map = output.detach().cpu().numpy().reshape(output.shape[2], output.shape[3])
     density_map_color = plt.cm.jet(density_map / np.max(density_map))
+
     return predicted_count, density_map_color
+from gradio.inputs import Image
+from gradio.outputs import Label, Image
+input_interface = gr.inputs.Image(label="Input Image")
+output_interface = [
+    Label(label="Predicted Count"),
+    Image(label="Density Map", type="numpy")
+]
 
-output_interface = gr.outputs.Textbox(label="Predicted Count")
-density_map_interface = gr.outputs.Image(label="Density Map")
+# Create the Gradio app with both interfaces
+grapp = gr.Interface(fn=predict_count, inputs=input_interface, outputs=output_interface)
 
-grapp = gr.Interface(fn=predict_count, inputs=input_interface, outputs=[output_interface, density_map_interface])
 # Launch the app
 grapp.launch()