Update app.py

app.py

@@ -1,73 +1,3 @@
-#Install CLIP-Interrogator-2.1
-
-import os, subprocess
-import torch
-
-def setup():
-    install_cmds = [
-        ['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'))
-
-import sys
-sys.path.append('src/blip')
-sys.path.append('clip-interrogator')
-
-import gradio as gr
-from clip_interrogator import Config, Interrogator
-
-config = Config()
-config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
-config.blip_offload = False if torch.cuda.is_available() else True
-config.chunk_size = 2048
-config.flavor_intermediate_count = 512
-config.blip_num_beams = 64
-
-ci = Interrogator(config)
-
-def img_to_text(image, mode, best_max_flavors):
-    image = image.convert('RGB')
-    if mode == 'best':
-        
-        prompt_result = ci.interrogate(image, max_flavors=int(best_max_flavors))
-        
-        print("mode best: " + prompt_result)
-        
-        return prompt_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
-    
-    elif mode == 'classic':
-        
-        prompt_result = ci.interrogate_classic(image)
-        
-        print("mode classic: " + prompt_result)
-        
-        return prompt_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
-    
-    else:
-        
-        prompt_result = ci.interrogate_fast(image)
-        
-        print("mode fast: " + prompt_result)
-        
-        return prompt_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
-
 
 
 from PIL import Image
@@ -97,7 +27,7 @@ import mutagen
 from mutagen.mp3 import MP3
 from mutagen.wave import WAVE
 
-#img_to_text = gr.Blocks.load(name="spaces/fffiloni/CLIP-Interrogator-2")
+img_to_text = gr.Blocks.load(name="spaces/fffiloni/CLIP-Interrogator-2")
 text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music") 
 
 language_translation_model = hub.Module(name='baidu_translate')