Update app.py

app.py

@@ -167,22 +167,23 @@ def infer(audio_in, oauth_token: gr.OAuthToken):
 
     gr.Info("Calling SALMONN to understand audio...")
     salmonn_res = get_salmonn(audio_in, salmonn_prompt, gradio_auth_token)
-    yield None, salmonn_res, None, None
+    yield None, salmonn_res, None
 
     gr.Info("Creating an outfit suggestion based on audio understanding...")
     outfit_sentence = get_outfit_prompt(salmonn_res)
-    yield outfit_sentence, salmonn_res, None, None
+    yield outfit_sentence, salmonn_res, None
 
     gr.Info("Generate an image with SDXL Lightning...")
     outfit_image = sdxl_image(outfit_sentence, gradio_auth_token)
 
+    """
     gr.Info("Get outfit parts...")
     item_prompts = get_parsed_outfit_items(outfit_sentence)
 
     gr.Info("Generate shopping gallery...")
     images_with_labels = generate_sdxl_images_list_dynamic(item_prompts, gradio_auth_token)
-    
-    yield outfit_sentence, salmonn_res, outfit_image, images_with_labels
+    """
+    yield outfit_sentence, salmonn_res, outfit_image
 
 demo_title = "Music to Outfit"
 description = "Get an outfit idea from audio/music input"
@@ -239,7 +240,7 @@ with gr.Blocks(css=css) as demo:
                     label = "Outfit propal"
                 )   
 
-                clothes_gallery = gr.Gallery()
+                #clothes_gallery = gr.Gallery()
     
     submit_btn.click(
         fn = infer,
@@ -250,7 +251,7 @@ with gr.Blocks(css=css) as demo:
             caption,
             salmonn_desc,
             result,
-            clothes_gallery
+            #clothes_gallery
         ]
     )