Update app.py

app.py

@@ -6,13 +6,10 @@ from clip_interrogator import Config, Interrogator
 # MODELS = ['ViT-L (best for Stable Diffusion 1.*)',]
 
 # load BLIP and ViT-L https://huggingface.co/openai/clip-vit-large-patch14
-config = Config(clip_model_name="ViT-L-14/openai")
+from PIL import Image
+from clip_interrogator import Config, Interrogator
 
-ci = Interrogator(config)
-ci.clip_model = ci_vitl.clip_model.to("cpu")
-ci.config.blip_num_beams = 64
-ci.config.chunk_size = 2048
-ci.config.flavor_intermediate_count = 2048 #  1024
+ci = Interrogator(Config(clip_model_name="ViT-L-14/openai"))
 
 def image_analysis(image):
     image = image.convert('RGB')
@@ -33,7 +30,7 @@ def image_analysis(image):
     return medium_ranks, artist_ranks, movement_ranks, trending_ranks, flavor_ranks
 
 
-def image_to_prompt(image, clip_model_name, mode):
+def image_to_prompt(image, mode):
     image = image.convert('RGB')
     if mode == 'best':
         prompt = ci.interrogate(image)
@@ -129,7 +126,7 @@ def analyze_tab():
             trending = gr.Label(label="Trending", num_top_classes=5)
             flavor = gr.Label(label="Flavor", num_top_classes=5)
 
-    button = gr.Button("Analyze", api_name="image-analysis")
+    button = gr.Button("Analyze")
     button.click(image_analysis, inputs=[image, model], outputs=[medium, artist, movement, trending, flavor])
 
     examples=[['example01.jpg', MODELS[0]], ['example02.jpg', MODELS[0]]]
@@ -153,14 +150,14 @@ with gr.Blocks(css=CSS) as block:
                 input_image = gr.Image(type='pil', elem_id="input-img")
                 with gr.Column():
                     input_mode = gr.Radio(['best', 'fast', 'classic', 'negative'], value='best', label='Mode')
-            submit_btn = gr.Button("Submit", api_name="image-to-prompt")
+            submit_btn = gr.Button("Submit")
             output_text = gr.Textbox(label="Output", elem_id="output-txt")
 
             examples=[['example01.jpg', MODELS[0], 'best'], ['example02.jpg', MODELS[0], 'best']]
             ex = gr.Examples(
                 examples=examples, 
                 fn=image_to_prompt, 
-                inputs=[input_image, input_model, input_mode], 
+                inputs=[input_image, input_mode], 
                 outputs=[output_text], 
                 cache_examples=True, 
                 run_on_click=True
@@ -174,7 +171,7 @@ with gr.Blocks(css=CSS) as block:
 
     submit_btn.click(
         fn=image_to_prompt, 
-        inputs=[input_image, input_model, input_mode], 
+        inputs=[input_image, input_mode], 
         outputs=[output_text]
     )
     share_button.click(None, [], [], _js=share_js)