Update

app_cerebras.py

@@ -11,7 +11,7 @@ demo = get_app(
         "llama3.1-405b",
     ],
     default_model="llama3.1-70b",
-    registry=cerebras_gradio.registry,
+    src=cerebras_gradio.registry,
     accept_token=not os.getenv("CEREBRAS_API_KEY"),
 )
 

app_claude.py

@@ -13,7 +13,7 @@ demo = get_app(
         "claude-3-haiku-20240307",
     ],
     default_model="claude-3-5-sonnet-20241022",
-    registry=anthropic_gradio.registry,
+    src=anthropic_gradio.registry,
     accept_token=not os.getenv("ANTHROPIC_API_KEY"),
 )
 

app_fireworks.py

@@ -10,7 +10,7 @@ demo = get_app(
         "f1-mini-preview",
     ],
     default_model="f1-preview",
-    registry=fireworks_gradio.registry,
+    src=fireworks_gradio.registry,
     accept_token=not os.getenv("FIREWORKS_API_KEY"),
 )
 

app_gemini.py

@@ -12,7 +12,7 @@ demo = get_app(
         "gemini-exp-1114",
     ],
     default_model="gemini-1.5-pro",
-    registry=gemini_gradio.registry,
+    src=gemini_gradio.registry,
     accept_token=not os.getenv("GEMINI_API_KEY"),
 )
 

app_groq.py

@@ -17,7 +17,7 @@ demo = get_app(
         "gemma-7b-it",
     ],
     default_model="llama3-groq-70b-8192-tool-use-preview",
-    registry=groq_gradio.registry,
+    src=groq_gradio.registry,
     accept_token=not os.getenv("GROQ_API_KEY"),
 )
 

app_hf.py

@@ -21,7 +21,7 @@ demo = get_app(
         "openai-community/gpt2",
     ],
     default_model="HuggingFaceTB/SmolLM2-1.7B-Instruct",
-    registry="models",
+    src="models",
 )
 
 if __name__ == "__main__":

app_hyperbolic.py

@@ -17,7 +17,7 @@ demo = get_app(
         "meta-llama/Meta-Llama-3.1-405B-Instruct",
     ],
     default_model="Qwen/Qwen2.5-Coder-32B-Instruct",
-    registry=hyperbolic_gradio.registry,
+    src=hyperbolic_gradio.registry,
     accept_token=not os.getenv("HYPERBOLIC_API_KEY"),
 )
 

app_mistral.py

@@ -19,7 +19,7 @@ demo = get_app(
         "open-codestral-mamba",
     ],
     default_model="pixtral-large-latest",
-    registry=mistral_gradio.registry,
+    src=mistral_gradio.registry,
     accept_token=not os.getenv("MISTRAL_API_KEY"),
 )
 

app_nvidia.py

@@ -44,7 +44,7 @@ demo = get_app(
         "snowflake/arctic",
     ],
     default_model="meta/llama-3.1-70b-instruct",
-    registry=nvidia_gradio.registry,
+    src=nvidia_gradio.registry,
     accept_token=not os.getenv("NVIDIA_API_KEY"),
 )
 

app_openai.py

@@ -26,7 +26,7 @@ demo = get_app(
         "gpt-4-0613",
     ],
     default_model="gpt-4o-2024-11-20",
-    registry=openai_gradio.registry,
+    src=openai_gradio.registry,
     accept_token=not os.getenv("OPENAI_API_KEY"),
 )
 

app_perplexity.py

@@ -15,7 +15,7 @@ demo = get_app(
         "llama-3.1-70b-instruct",
     ],
     default_model="llama-3.1-sonar-large-128k-online",
-    registry=perplexity_gradio.registry,
+    src=perplexity_gradio.registry,
     accept_token=not os.getenv("PERPLEXITY_API_KEY"),
 )
 

app_qwen.py

@@ -17,7 +17,7 @@ demo = get_app(
         "qwen1.5-7b-chat",
     ],
     default_model="qwen-turbo-latest",
-    registry=dashscope_gradio.registry,
+    src=dashscope_gradio.registry,
     accept_token=not os.getenv("DASHSCOPE_API_KEY"),
 )
 

app_sambanova.py

@@ -15,7 +15,7 @@ demo = get_app(
         "Meta-Llama-3.1-405B-Instruct",
     ],
     default_model="Llama-3.2-90B-Vision-Instruct",
-    registry=sambanova_gradio.registry,
+    src=sambanova_gradio.registry,
     accept_token=not os.getenv("SAMBANOVA_API_KEY"),
     multimodal=True,
 )

app_together.py

@@ -42,7 +42,7 @@ demo = get_app(
         "upstage/SOLAR-10.7B-Instruct-v1.0",
     ],
     default_model="meta-llama/Llama-3.2-11B-Vision-Instruct-Turbo",
-    registry=together_gradio.registry,
+    src=together_gradio.registry,
     accept_token=not os.getenv("TOGETHER_API_KEY"),
     multimodal=True,
 )

app_xai.py

@@ -10,7 +10,7 @@ demo = get_app(
         "grok-vision-beta",
     ],
     default_model="grok-vision-beta",
-    registry=xai_gradio.registry,
+    src=xai_gradio.registry,
     accept_token=not os.getenv("XAI_API_KEY"),
 )
 

utils.py

@@ -1,4 +1,4 @@
-from typing import Callable
+from typing import Callable, Literal
 
 import gradio as gr
 
@@ -6,7 +6,7 @@ import gradio as gr
 def get_app(
     models: list[str],
     default_model: str,
-    registry: Callable,
+    src: Callable[[str, str | None], gr.Blocks] | Literal["models"],
     accept_token: bool = False,
     **kwargs,
 ) -> gr.Blocks:
@@ -19,7 +19,7 @@ def get_app(
         columns = []
         for model_name in models:
             with gr.Column(visible=model_name == default_model) as column:
-                gr.load(name=model_name, src=registry, accept_token=accept_token, **kwargs)
+                gr.load(name=model_name, src=src, accept_token=accept_token, **kwargs)
             columns.append(column)
 
         model.change(