Update app.py

Removed some models and added some others

app.py

@@ -20,17 +20,15 @@ endpoint_base_url = "https://api-inference.huggingface.co/models/"
 endpoints = [
     "Salesforce/blip-image-captioning-large",
     "Salesforce/blip-image-captioning-base",
-    "Salesforce/blip2-opt-2.7b",
-    "Salesforce/instructblip-vicuna-7b",
-    "Salesforce/instructblip-flan-t5-xl",
-    "facebook/nougat-base",
     "nlpconnect/vit-gpt2-image-captioning",
-    "jinhybr/OCR-Donut-CORD",
     "microsoft/git-base",
+    "microsoft/kosmos-2-patch14-224",
+    "microsoft/git-large-textcaps",
+    "microsoft/git-large-r-coco",
+    "facebook/nougat-base",
     "Norm/nougat-latex-base",
+    "bipin/image-caption-generator",
 ]
-## Test later:
-# "microsoft/kosmos-2-patch14-224",
 
 
 # --- Define helper functions ---
@@ -77,13 +75,12 @@ def captioner(image):
     results = get_completion(base64_image)
     captions = []
     for endpoint, result in results.items():
-        model_name = endpoint.split("/")[-1]  # Extract the model name from the endpoint
         if "error" not in result:
             caption = (
-                f"**{model_name.upper()}**: \n {result[0]['generated_text']} \n\n\n "
+                f"**{endpoint}**: \n\n {result[0]['generated_text']} \n\n\n "
             )
         else:
-            caption = f"**{model_name.upper()}**: \n Error - {result['error']} \n\n\n "
+            caption = f"**{model_name.upper()}**: \n\n Error - {result['error']} \n\n\n "
         captions.append(caption)
     return "".join(captions)  # Join all captions into a single string