Update app.py

app.py

@@ -9,6 +9,7 @@ from gradio_client.client import Job
 
 DESCRIPTION = "# Comparing image captioning models"
 ORIGINAL_SPACE_INFO = """\
+- [BLIP fine-tuned for long captions](https://huggingface.co/spaces/unography/image-captioning-with-longcap)
 - [GIT-large fine-tuned on COCO](https://huggingface.co/spaces/library-samples/image-captioning-with-git)
 - [BLIP-large](https://huggingface.co/spaces/library-samples/image-captioning-with-blip)
 - [BLIP-2 OPT 6.7B](https://huggingface.co/spaces/merve/BLIP2-with-transformers)
@@ -26,6 +27,15 @@ torch.hub.download_url_to_file(
 )
 
 
+def generate_caption_longcap(image_path: str, return_job: bool = False) -> str | Job:
+    try:
+        client = Client("hysts/image-captioning-with-git")
+        fn = client.submit if return_job else client.predict
+        return fn(image_path, api_name="/caption")
+    except Exception:
+        gr.Warning("The GIT-large Space is currently unavailable. Please try again later.")
+        return ""
+
 def generate_caption_git(image_path: str, return_job: bool = False) -> str | Job:
     try:
         client = Client("hysts/image-captioning-with-git")
@@ -116,8 +126,9 @@ def generate_caption_fuyu(image_path: str, return_job: bool = False) -> str | Jo
         return ""
 
 
-def generate_captions(image_path: str) -> tuple[str, str, str, str, str, str]:
+def generate_captions(image_path: str) -> tuple[str, str, str, str, str, str, str]:
     jobs = [
+        generate_caption_longcap(image_path, return_job=True),
         generate_caption_git(image_path, return_job=True),
         generate_caption_blip(image_path, return_job=True),
         generate_caption_blip2_opt(image_path, return_job=True),
@@ -135,6 +146,7 @@ with gr.Blocks(css="style.css") as demo:
             input_image = gr.Image(type="filepath")
             run_button = gr.Button("Caption")
         with gr.Column():
+            out_longcap = gr.Textbox(label="BLIP fine-tuned for long captions")
             out_git = gr.Textbox(label="GIT-large fine-tuned on COCO")
             out_blip = gr.Textbox(label="BLIP-large")
             out_blip2_opt = gr.Textbox(label="BLIP-2 OPT 6.7B")
@@ -143,6 +155,7 @@ with gr.Blocks(css="style.css") as demo:
             out_fuyu = gr.Textbox(label="Fuyu-8B")
 
     outputs = [
+        out_longcap,
         out_git,
         out_blip,
         out_blip2_opt,