Update app.py

app.py

@@ -16,7 +16,7 @@ git_processor_large_coco = AutoProcessor.from_pretrained("microsoft/git-large-co
 git_model_large_coco = AutoModelForCausalLM.from_pretrained("microsoft/git-large-coco")
 
 git_processor_large_textcaps = AutoProcessor.from_pretrained("microsoft/git-large-r-textcaps")
-git_model_large_textcaps = AutoModelForCausalLM.from_pretrained("microsoft/git-large-r-textcaps")
+# git_model_large_textcaps = AutoModelForCausalLM.from_pretrained("microsoft/git-large-r-textcaps")
 
 # blip_processor_base = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 # blip_model_base = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
@@ -44,7 +44,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # git_model_base.to(device)
 # blip_model_base.to(device)
 git_model_large_coco.to(device)
-git_model_large_textcaps.to(device)
+# git_model_large_textcaps.to(device)
 blip_model_large.to(device)
 # vitgpt_model.to(device)
 coca_model.to(device)
@@ -78,7 +78,7 @@ def generate_captions(image):
 
     caption_git_large_coco = generate_caption(git_processor_large_coco, git_model_large_coco, image)
 
-    caption_git_large_textcaps = generate_caption(git_processor_large_textcaps, git_model_large_textcaps, image)
+    # caption_git_large_textcaps = generate_caption(git_processor_large_textcaps, git_model_large_textcaps, image)
 
     # caption_blip_base = generate_caption(blip_processor_base, blip_model_base, image)
 
@@ -92,7 +92,7 @@ def generate_captions(image):
 
     caption_blip2_8_bit = generate_caption(blip2_processor_8_bit, blip2_model_8_bit, image, use_float_16=True).strip()
 
-    return caption_git_large_coco, caption_git_large_textcaps, caption_blip_large, caption_coca, caption_blip2_8_bit
+    return caption_git_large_coco, caption_blip_large, caption_coca, caption_blip2_8_bit
 
    
 examples = [["cats.jpg"], ["stop_sign.png"], ["astronaut.jpg"]]