update provider label
Browse files
app.py
CHANGED
|
@@ -16,21 +16,19 @@ from app_gemini import demo as demo_gemini
|
|
| 16 |
from app_groq import demo as demo_groq
|
| 17 |
from app_hyperbolic import demo as demo_hyperbolic
|
| 18 |
from app_fal import demo as demo_fal
|
| 19 |
-
#from app_marco_o1 import demo as demo_marco_o1
|
| 20 |
from app_mistral import demo as demo_mistral
|
| 21 |
from app_nvidia import demo as demo_nvidia
|
| 22 |
from app_openai import demo as demo_openai
|
| 23 |
from app_perplexity import demo as demo_perplexity
|
| 24 |
from app_qwen import demo as demo_qwen
|
| 25 |
-
#from app_sailor import demo as demo_sailor
|
| 26 |
from app_sambanova import demo as demo_sambanova
|
| 27 |
from app_together import demo as demo_together
|
| 28 |
from app_xai import demo as demo_grok
|
| 29 |
from app_showui import demo as demo_showui
|
| 30 |
from app_omini import demo as demo_omini
|
| 31 |
|
| 32 |
-
# Create mapping of
|
| 33 |
-
|
| 34 |
"Gemini": demo_gemini,
|
| 35 |
"Grok": demo_grok,
|
| 36 |
"Cohere": demo_cohere,
|
|
@@ -55,15 +53,14 @@ DEMOS = {
|
|
| 55 |
"Perplexity": demo_perplexity,
|
| 56 |
"Experimental": demo_experimental,
|
| 57 |
"Mistral": demo_mistral,
|
| 58 |
-
"NVIDIA": demo_nvidia
|
| 59 |
-
#"Marco-o1": demo_marco_o1,
|
| 60 |
-
#"Sailor": demo_sailor,
|
| 61 |
}
|
| 62 |
|
| 63 |
demo = get_app(
|
| 64 |
-
models=list(
|
| 65 |
default_model="Gemini",
|
| 66 |
-
src=
|
|
|
|
| 67 |
)
|
| 68 |
|
| 69 |
if __name__ == "__main__":
|
|
|
|
| 16 |
from app_groq import demo as demo_groq
|
| 17 |
from app_hyperbolic import demo as demo_hyperbolic
|
| 18 |
from app_fal import demo as demo_fal
|
|
|
|
| 19 |
from app_mistral import demo as demo_mistral
|
| 20 |
from app_nvidia import demo as demo_nvidia
|
| 21 |
from app_openai import demo as demo_openai
|
| 22 |
from app_perplexity import demo as demo_perplexity
|
| 23 |
from app_qwen import demo as demo_qwen
|
|
|
|
| 24 |
from app_sambanova import demo as demo_sambanova
|
| 25 |
from app_together import demo as demo_together
|
| 26 |
from app_xai import demo as demo_grok
|
| 27 |
from app_showui import demo as demo_showui
|
| 28 |
from app_omini import demo as demo_omini
|
| 29 |
|
| 30 |
+
# Create mapping of providers to their demos
|
| 31 |
+
PROVIDERS = {
|
| 32 |
"Gemini": demo_gemini,
|
| 33 |
"Grok": demo_grok,
|
| 34 |
"Cohere": demo_cohere,
|
|
|
|
| 53 |
"Perplexity": demo_perplexity,
|
| 54 |
"Experimental": demo_experimental,
|
| 55 |
"Mistral": demo_mistral,
|
| 56 |
+
"NVIDIA": demo_nvidia
|
|
|
|
|
|
|
| 57 |
}
|
| 58 |
|
| 59 |
demo = get_app(
|
| 60 |
+
models=list(PROVIDERS.keys()),
|
| 61 |
default_model="Gemini",
|
| 62 |
+
src=PROVIDERS,
|
| 63 |
+
dropdown_label="Select Provider"
|
| 64 |
)
|
| 65 |
|
| 66 |
if __name__ == "__main__":
|
utils.py
CHANGED
|
@@ -8,13 +8,14 @@ def get_app(
|
|
| 8 |
default_model: str,
|
| 9 |
src: Union[Callable[[str, str | None], gr.Blocks], Literal["models"], Dict[str, gr.Blocks]],
|
| 10 |
accept_token: bool = False,
|
|
|
|
| 11 |
**kwargs,
|
| 12 |
) -> gr.Blocks:
|
| 13 |
def update_model(new_model: str) -> list[gr.Column]:
|
| 14 |
return [gr.Column(visible=model_name == new_model) for model_name in models]
|
| 15 |
|
| 16 |
with gr.Blocks() as demo:
|
| 17 |
-
model = gr.Dropdown(label=
|
| 18 |
|
| 19 |
columns = []
|
| 20 |
for model_name in models:
|
|
|
|
| 8 |
default_model: str,
|
| 9 |
src: Union[Callable[[str, str | None], gr.Blocks], Literal["models"], Dict[str, gr.Blocks]],
|
| 10 |
accept_token: bool = False,
|
| 11 |
+
dropdown_label: str = "Select Model",
|
| 12 |
**kwargs,
|
| 13 |
) -> gr.Blocks:
|
| 14 |
def update_model(new_model: str) -> list[gr.Column]:
|
| 15 |
return [gr.Column(visible=model_name == new_model) for model_name in models]
|
| 16 |
|
| 17 |
with gr.Blocks() as demo:
|
| 18 |
+
model = gr.Dropdown(label=dropdown_label, choices=models, value=default_model)
|
| 19 |
|
| 20 |
columns = []
|
| 21 |
for model_name in models:
|