simplifying imports

app.py

@@ -3,12 +3,9 @@ from idc_index import index
 from pathlib import Path
 import pydicom
 import pandas as pd
-# import pyarrow as pa
-# import pyarrow.parquet as pq
 from tempfile import TemporaryDirectory
 import os
 from pathlib import Path
-# import polars
 import pydicom.datadict as dd
 import shutil
 import papermill as pm 
@@ -65,11 +62,13 @@ if st.button("Run inference"):
 
     subprocess.run(["wget", "https://raw.githubusercontent.com/deepakri201/DICOMScanClassification_pw41/main/DICOMScanClassification_user_demo.ipynb"])
 
-    pm.execute_notebook(
-            "DICOMScanClassification_user_demo.ipynb",
-            'output.ipynb',
-            parameters = dict(SeriesInstanceUID=selected_series)
-    )
+    # pm.execute_notebook(
+    #         "DICOMScanClassification_user_demo.ipynb",
+    #         'output.ipynb',
+    #         parameters = dict(SeriesInstanceUID=selected_series)
+    # )
+    
+    subprocess.run(["papermill", "-p", "SeriesInstanceUID", selected_series,  "DICOMScanClassification_user_demo.ipynb", "output.ipynb"])
 
     with open('output.ipynb', "rb") as f:
         st.download_button(