Update app.py

app.py

@@ -7,7 +7,7 @@ import subprocess
 subprocess_executed = False
 
 # Function to install dependencies
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def install_dependencies():
     command = "chmod +x install.sh"
     subprocess.run(command, shell=True, check=True)
@@ -16,7 +16,7 @@ def install_dependencies():
     subprocess.run(command, shell=True, check=True)
 
 # Function to run inference
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def run_inference():
     command = "chmod +x inference.sh"
     subprocess.run(command, shell=True, check=True)
@@ -76,6 +76,15 @@ def main():
                         if os.path.exists(video_path) and os.path.exists(image_path) and os.path.exists(largest_slice):
                             zip_filename = os.path.join(temp_dir, "outputs.zip")
                             zip_and_download(temp_dicom_dir_path, zip_filename)
+
+                            st.title("Largest Slice")
+                            st.image(largest_slice, use_column_width=True)
+
+                            st.title("Video")
+                            st.video(video_path, format="video/mp4")
+
+                            st.title("Diameter Graph")
+                            st.image(image_path, use_column_width=True)
                         else:
                             st.error("Output files not found.")
                     else: