Update app.py

app.py

@@ -102,34 +102,6 @@ def create_mask_zip(mask, output_name='masks.zip'):
     buffer.seek(0)
     return buffer, output_name
 
-class MaskZipOutput(gr.outputs.OutputComponent):
-    def __init__(self):
-        self.label = "Download Masks"
-        self.filename = "masks.zip"
-        self.mime_type = "application/zip"
-
-    def get_template_context(self):
-        return {
-            "label": self.label,
-            "filename": self.filename,
-            "mime_type": self.mime_type
-        }
-
-    def component_type(self):
-        return "output.file"
-
-    def to_response(self, value):
-        if value is None:
-            return {}
-        return {
-            "file": {
-                "name": self.filename,
-                "content": value.read(),
-                "mime_type": self.mime_type
-            }
-        }
-
-
 # Define the custom legend HTML
 legend_html = '''
 <div style="font-size: 14px; font-weight: bold; display: flex; flex-wrap: wrap; margin-top: 10px;">