Spaces:
Running
refactor(provider): unify default provider constant
Browse files- [refactor] Remove `DEFAULT_IMAGE_PROVIDER`, `DEFAULT_IMAGE_TO_IMAGE_PROVIDER`, `DEFAULT_TTS_PROVIDER` constants (utils.py:14-16)
- [refactor] Add `DEFAULT_PROVIDER` constant set to "auto" (utils.py:19)
- [docs] Add comment for `DEFAULT_PROVIDER` (utils.py:20)
- [refactor] Update imports for provider constants (ui_components.py:6-9)
- [update] Set `chat_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:52)
- [update] Set `img_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:194)
- [update] Set `img2img_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:314)
- [update] Set `tts_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:455)
- [docs] Remove "(default for images)" from `fal-ai` description in footer (ui_components.py:691)
- ui_components.py +8 -8
- utils.py +3 -3
|
@@ -5,9 +5,9 @@ Contains functions to create different sections of the Gradio interface.
|
|
| 5 |
|
| 6 |
import gradio as gr
|
| 7 |
from utils import (
|
| 8 |
-
DEFAULT_CHAT_MODEL, DEFAULT_IMAGE_MODEL,
|
| 9 |
-
DEFAULT_IMAGE_TO_IMAGE_MODEL,
|
| 10 |
-
DEFAULT_TTS_MODEL,
|
| 11 |
CHAT_CONFIG, IMAGE_CONFIG, IMAGE_PROVIDERS, IMAGE_MODEL_PRESETS,
|
| 12 |
IMAGE_TO_IMAGE_MODEL_PRESETS, TTS_MODEL_PRESETS, TTS_VOICES, TTS_MODEL_CONFIGS,
|
| 13 |
IMAGE_EXAMPLE_PROMPTS, IMAGE_TO_IMAGE_EXAMPLE_PROMPTS, TTS_EXAMPLE_TEXTS, TTS_EXAMPLE_AUDIO_URLS
|
|
@@ -49,7 +49,7 @@ def create_chat_tab(handle_chat_submit_fn, handle_chat_retry_fn=None):
|
|
| 49 |
)
|
| 50 |
chat_provider = gr.Dropdown(
|
| 51 |
choices=IMAGE_PROVIDERS,
|
| 52 |
-
value=
|
| 53 |
label="Provider",
|
| 54 |
interactive=True
|
| 55 |
)
|
|
@@ -191,7 +191,7 @@ def create_image_tab(handle_image_generation_fn):
|
|
| 191 |
)
|
| 192 |
img_provider = gr.Dropdown(
|
| 193 |
choices=IMAGE_PROVIDERS,
|
| 194 |
-
value=
|
| 195 |
label="Provider",
|
| 196 |
interactive=True
|
| 197 |
)
|
|
@@ -311,7 +311,7 @@ def create_image_to_image_tab(handle_image_to_image_generation_fn):
|
|
| 311 |
)
|
| 312 |
img2img_provider = gr.Dropdown(
|
| 313 |
choices=IMAGE_PROVIDERS,
|
| 314 |
-
value=
|
| 315 |
label="Provider",
|
| 316 |
interactive=True
|
| 317 |
)
|
|
@@ -452,7 +452,7 @@ def create_tts_tab(handle_tts_generation_fn):
|
|
| 452 |
)
|
| 453 |
tts_provider = gr.Dropdown(
|
| 454 |
choices=IMAGE_PROVIDERS,
|
| 455 |
-
value=
|
| 456 |
label="Provider",
|
| 457 |
interactive=True
|
| 458 |
)
|
|
@@ -687,7 +687,7 @@ def create_footer():
|
|
| 687 |
- Supports both fal-ai and replicate providers
|
| 688 |
|
| 689 |
**Supported Providers:**
|
| 690 |
-
- **fal-ai**: High-quality image generation
|
| 691 |
- **hf-inference**: Core API with comprehensive model support
|
| 692 |
- **cerebras**: High-performance inference
|
| 693 |
- **cohere**: Advanced language models with multilingual support
|
|
|
|
| 5 |
|
| 6 |
import gradio as gr
|
| 7 |
from utils import (
|
| 8 |
+
DEFAULT_CHAT_MODEL, DEFAULT_IMAGE_MODEL, DEFAULT_PROVIDER,
|
| 9 |
+
DEFAULT_IMAGE_TO_IMAGE_MODEL,
|
| 10 |
+
DEFAULT_TTS_MODEL,
|
| 11 |
CHAT_CONFIG, IMAGE_CONFIG, IMAGE_PROVIDERS, IMAGE_MODEL_PRESETS,
|
| 12 |
IMAGE_TO_IMAGE_MODEL_PRESETS, TTS_MODEL_PRESETS, TTS_VOICES, TTS_MODEL_CONFIGS,
|
| 13 |
IMAGE_EXAMPLE_PROMPTS, IMAGE_TO_IMAGE_EXAMPLE_PROMPTS, TTS_EXAMPLE_TEXTS, TTS_EXAMPLE_AUDIO_URLS
|
|
|
|
| 49 |
)
|
| 50 |
chat_provider = gr.Dropdown(
|
| 51 |
choices=IMAGE_PROVIDERS,
|
| 52 |
+
value=DEFAULT_PROVIDER,
|
| 53 |
label="Provider",
|
| 54 |
interactive=True
|
| 55 |
)
|
|
|
|
| 191 |
)
|
| 192 |
img_provider = gr.Dropdown(
|
| 193 |
choices=IMAGE_PROVIDERS,
|
| 194 |
+
value=DEFAULT_PROVIDER,
|
| 195 |
label="Provider",
|
| 196 |
interactive=True
|
| 197 |
)
|
|
|
|
| 311 |
)
|
| 312 |
img2img_provider = gr.Dropdown(
|
| 313 |
choices=IMAGE_PROVIDERS,
|
| 314 |
+
value=DEFAULT_PROVIDER,
|
| 315 |
label="Provider",
|
| 316 |
interactive=True
|
| 317 |
)
|
|
|
|
| 452 |
)
|
| 453 |
tts_provider = gr.Dropdown(
|
| 454 |
choices=IMAGE_PROVIDERS,
|
| 455 |
+
value=DEFAULT_PROVIDER,
|
| 456 |
label="Provider",
|
| 457 |
interactive=True
|
| 458 |
)
|
|
|
|
| 687 |
- Supports both fal-ai and replicate providers
|
| 688 |
|
| 689 |
**Supported Providers:**
|
| 690 |
+
- **fal-ai**: High-quality image generation
|
| 691 |
- **hf-inference**: Core API with comprehensive model support
|
| 692 |
- **cerebras**: High-performance inference
|
| 693 |
- **cohere**: Advanced language models with multilingual support
|
|
@@ -11,11 +11,11 @@ import requests
|
|
| 11 |
# Configuration constants
|
| 12 |
DEFAULT_CHAT_MODEL = "openai/gpt-oss-20b"
|
| 13 |
DEFAULT_IMAGE_MODEL = "Qwen/Qwen-Image"
|
| 14 |
-
DEFAULT_IMAGE_PROVIDER = "fal-ai"
|
| 15 |
DEFAULT_IMAGE_TO_IMAGE_MODEL = "Qwen/Qwen-Image-Edit"
|
| 16 |
-
DEFAULT_IMAGE_TO_IMAGE_PROVIDER = "fal-ai"
|
| 17 |
DEFAULT_TTS_MODEL = "hexgrad/Kokoro-82M"
|
| 18 |
-
|
|
|
|
|
|
|
| 19 |
|
| 20 |
# Chat configuration
|
| 21 |
CHAT_CONFIG = {
|
|
|
|
| 11 |
# Configuration constants
|
| 12 |
DEFAULT_CHAT_MODEL = "openai/gpt-oss-20b"
|
| 13 |
DEFAULT_IMAGE_MODEL = "Qwen/Qwen-Image"
|
|
|
|
| 14 |
DEFAULT_IMAGE_TO_IMAGE_MODEL = "Qwen/Qwen-Image-Edit"
|
|
|
|
| 15 |
DEFAULT_TTS_MODEL = "hexgrad/Kokoro-82M"
|
| 16 |
+
|
| 17 |
+
# Unified default provider used by all non-chat tasks
|
| 18 |
+
DEFAULT_PROVIDER = "auto"
|
| 19 |
|
| 20 |
# Chat configuration
|
| 21 |
CHAT_CONFIG = {
|