Update app.py

app.py

@@ -3,30 +3,7 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 import os, subprocess
 import torch
 
-# def setup():
-#     install_cmds = [
-#         ['pip', 'install', 'ftfy', 'gradio', 'regex', 'tqdm', 'transformers==4.21.2', 'timm', 'fairscale', 'requests'],
-#         ['pip', 'install', 'open_clip_torch'],
-#         ['pip', 'install', '-e', 'git+https://github.com/pharmapsychotic/BLIP.git@lib#egg=blip'],
-#         ['git', 'clone', '-b', 'open-clip', 'https://github.com/pharmapsychotic/clip-interrogator.git']
-#     ]
-#     for cmd in install_cmds:
-#         print(subprocess.run(cmd, stdout=subprocess.PIPE).stdout.decode('utf-8'))
-
-# setup()
-
-# download cache files
-# print("Download preprocessed cache files...")
-# CACHE_URLS = [
-#     'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_artists.pkl',
-#     'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_flavors.pkl',
-#     'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_mediums.pkl',
-#     'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_movements.pkl',
-#     'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_trendings.pkl',
-# ]
-# os.makedirs('cache', exist_ok=True)
-# for url in CACHE_URLS:
-#     print(subprocess.run(['wget', url, '-P', 'cache'], stdout=subprocess.PIPE).stdout.decode('utf-8'))
+# (Code for setup and downloading cache files)
 
 import sys
 sys.path.append('src/blip')
@@ -59,8 +36,6 @@ def inference(input_images, mode, best_max_flavors):
         prompt_texts.append(prompt_result['prompt'])  # Extract the prompt text
     return prompt_texts
 
-
-
 title = """
     <div style="text-align: center; max-width: 500px; margin: 0 auto;">
         <div
@@ -149,8 +124,8 @@ a {text-decoration-line: underline; font-weight: 600;}
     height: 100% !important;
     font-size: 17px !important;
 }
-
 '''
+
 with gr.Blocks(css=css) as block:
     with gr.Column(elem_id="col-container"):
         gr.HTML(title)
@@ -163,9 +138,11 @@ with gr.Blocks(css=css) as block:
         submit_btn = gr.Button("Submit")
         
         # Display the prompt texts
-        gallery = gr.Gallery(
-            label="Outputs", show_label=True, elem_id="gallery", object_fit="contain", height="auto",
-            src = "data:application/json,{{outputs['prompt_texts']}}"
+        gallery = gr.HTML(
+            src="{{' '.join(outputs['prompt_texts'])}}",
+            type="div",
+            label="Outputs",
+            elem_id="gallery"
         )
 
         with gr.Group(elem_id="share-btn-container"):
@@ -175,4 +152,3 @@ with gr.Blocks(css=css) as block:
     submit_btn.click(fn=inference, inputs=[input_image,mode_input,flavor_input], outputs=[gallery], api_name="clipi2")
 
 block.queue(max_size=32,concurrency_count=10).launch(show_api=False)
-