Spaces:

TRI-ML
/

vlm-demo

Paused

abalakrishnaTRI commited on Feb 14, 2024

Commit

6eb4de0

1 Parent(s): 1cd40da

fix naming

Files changed (1) hide show

serve/__init__.py CHANGED Viewed

@@ -6,19 +6,19 @@ MODEL_ID_TO_NAME = OrderedDict(
     [
         (
             "prism-dinosiglip+13b",
-            "PrismaticVLM 13B",
         ),
         (
             "prism-dinosiglip+7b",
-            "PrismaticVLM 7B",
         ),
         (
             "prism-dinosiglip-controlled+13b",
-            "PrismaticVLM 13B (Controlled)",
         ),
         (
             "prism-dinosiglip-controlled+7b",
-            "PrismaticVLM 7B (Controlled)",
         ),
         ("llava-v1.5-13b", "LLaVA 1.5 13B"),
         ("llava-v1.5-7b", "LLaVA 1.5 7B"),
@@ -34,4 +34,4 @@ INTERACTION_MODES_MAP = OrderedDict(
         ("Visual Question Answering", "vqa"),
         ("True/False Visual Question Answering", "true_false"),
     ]
-)

     [
         (
             "prism-dinosiglip+13b",
+            "Prism 13B",
         ),
         (
             "prism-dinosiglip+7b",
+            "Prism 7B",
         ),
         (
             "prism-dinosiglip-controlled+13b",
+            "Prism 13B (Controlled)",
         ),
         (
             "prism-dinosiglip-controlled+7b",
+            "Prism 7B (Controlled)",
         ),
         ("llava-v1.5-13b", "LLaVA 1.5 13B"),
         ("llava-v1.5-7b", "LLaVA 1.5 7B"),
         ("Visual Question Answering", "vqa"),
         ("True/False Visual Question Answering", "true_false"),
     ]
+)