Update app.py

app.py

@@ -154,8 +154,8 @@ def get_data(text):
     return results
 
 
-def send_to_img2img_def(image):
-    return image
+def send_to_img2img_def(images):
+    return images
 
 
 def send_to_txt2img(image):
@@ -407,6 +407,14 @@ with gr.Blocks(css=css) as demo:
 
             hf_text_button.click(hf_inference, inputs=[hf_prompt, hf_negative_prompt, hf_model, hf_steps, sampler, hf_cfg_scale, hf_width, hf_height,
                                                hf_seed], outputs=hf_image_output, concurrency_limit=64)
+        with gr.Tab("BLIP"):
+            with gr.Tab("Base"):
+                gr.load("models/Salesforce/blip-image-captioning-base", title="BLIP-base")
+            with gr.Tab("Large"):
+                gr.load("models/Salesforce/blip-image-captioning-large", title="BLIP-large")
+        with gr.Tab("Classification"):
+            gr.load("models/google/vit-base-patch16-224", title="ViT Classification")
+        with gr.Tab("Segmentation"):
+            gr.load("models/mattmdjaga/segformer_b2_clothes", title="SegFormer Segmentation")
             
-
 demo.queue(max_size=80, api_open=False).launch(max_threads=256, show_api=False)