Update app.py

app.py

@@ -39,11 +39,9 @@ def get_table_download_link(file_path):
     return href
 
 def CompressXML(xml_text):
-    tree = ET.ElementTree(ET.fromstring(xml_text))
-    for elem in tree.iter():
-        if isinstance(elem.tag, ET.Comment):
-            elem.getparent().remove(elem)
-    return ET.tostring(tree.getroot(), encoding='unicode')
+    root = ET.fromstring(xml_text)
+    ET.strip_elements(root, ET.Comment)
+    return ET.tostring(root, encoding='unicode')
 
 def read_file_content(file):
     if file.type == "application/json":
@@ -96,3 +94,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+