Added git base coco

app.py

@@ -16,7 +16,11 @@ model_blip_base = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-
 preprocessor_blip_large = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
 model_blip_large = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
 
-# Load the GIT coco model
+# Load the GIT coco base model
+preprocessor_git_base_coco = AutoProcessor.from_pretrained("microsoft/git-base-coco")
+model_git_base_coco = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco")
+
+# Load the GIT coco large model
 preprocessor_git_large_coco = AutoProcessor.from_pretrained("microsoft/git-large-coco")
 model_git_large_coco = AutoModelForCausalLM.from_pretrained("microsoft/git-large-coco")
 
@@ -30,6 +34,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Transfer the models to the device
 model_blip_base.to(device)
 model_blip_large.to(device)
+model_git_base_coco.to(device)
 model_git_large_coco.to(device)
 model_oc_coca.to(device)
 
@@ -142,8 +147,14 @@ def generate_captions(
 		caption_blip_large = generate_caption(preprocessor_blip_large, model_blip_large, image).strip()
 	except Exception as e:
 		print(e)
+
+	# Generate captions for the image using the GIT coco base model
+	try:
+		caption_git_base_coco = generate_caption(preprocessor_git_base_coco, model_git_base_coco, image).strip()
+	except Exception as e:
+		print(e)
 	
-	# Generate captions for the image using the GIT coco model
+	# Generate captions for the image using the GIT coco large model
 	try:
 		caption_git_large_coco = generate_caption(preprocessor_git_large_coco, model_git_large_coco, image).strip()
 	except Exception as e:
@@ -155,7 +166,7 @@ def generate_captions(
 	except Exception as e:
 		print(e)
 
-	return caption_blip_base, caption_blip_large, caption_git_large_coco, caption_oc_coca
+	return caption_blip_base, caption_blip_large, caption_git_base_coco, caption_git_large_coco, caption_oc_coca
 
 
 # Create the interface
@@ -171,6 +182,7 @@ iface = gr.Interface(
 	outputs=[
 		gr.outputs.Textbox(label="Blip base"),
 		gr.outputs.Textbox(label="Blip large"),
+		gr.outputs.Textbox(label="GIT base coco"),
 		gr.outputs.Textbox(label="GIT large coco"),
 		gr.outputs.Textbox(label="CLIP"),
 	],