AdritRao commited on
Commit
8c97be5
·
1 Parent(s): 3e93158

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,7 +54,7 @@ def run_inference():
54
  @st.cache_resource
55
  def convert_dicom_to_png(_dicom_data):
56
  # Convert DICOM to a PNG image
57
- plt.imshow(dicom_data.pixel_array, cmap=plt.cm.bone)
58
  plt.axis("off")
59
  plt.title("DICOM Image")
60
  plt.tight_layout()
@@ -88,7 +88,7 @@ if uploaded_zip_file is not None:
88
  dicom_data_list = [pydicom.dcmread(dicom_file) for dicom_file in dicom_files]
89
 
90
  # Convert DICOM slices to PNG frames
91
- png_frames = [convert_dicom_to_png(dicom_data) for dicom_data in dicom_data_list]
92
 
93
  # Create a list of image clips from PNG frames
94
  clips = [mp.ImageClip(png_frame, duration=1) for png_frame in png_frames]
 
54
  @st.cache_resource
55
  def convert_dicom_to_png(_dicom_data):
56
  # Convert DICOM to a PNG image
57
+ plt.imshow(_dicom_data.pixel_array, cmap=plt.cm.bone)
58
  plt.axis("off")
59
  plt.title("DICOM Image")
60
  plt.tight_layout()
 
88
  dicom_data_list = [pydicom.dcmread(dicom_file) for dicom_file in dicom_files]
89
 
90
  # Convert DICOM slices to PNG frames
91
+ png_frames = [convert_dicom_to_png(_dicom_data) for _dicom_data in dicom_data_list]
92
 
93
  # Create a list of image clips from PNG frames
94
  clips = [mp.ImageClip(png_frame, duration=1) for png_frame in png_frames]