Update app.py

app.py

@@ -485,11 +485,11 @@ def infer_seg(media, query):
 
     if not media:
         gr.Warning('Please upload an image')
-        return None, None, None
+        return None, None, None, None
 
     if not query:
         gr.Warning('Please provide a text prompt.')
-        return None, None, None
+        return None, None, None, None
 
     image = Image.open(media).convert('RGB')
     ori_width, ori_height = image.size
@@ -536,6 +536,7 @@ def infer_seg(media, query):
             answer,
             gr.update(value=None, visible=False),                      # hide AnnotatedImage
             gr.update(value=None, interactive=False, visible=False),   # hide DownloadButton
+            gr.update(value="", visible=True),
         )
     
     if len(quant_ids) % CODEBOOK_DEPTH != 0: