Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 722 Bytes
585c2f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import ai_gradio
from utils_ai_gradio import get_app
# Get the OpenAI models but keep their full names for loading
OPENAI_MODELS_FULL = [
k for k in ai_gradio.registry.keys()
if k.startswith('openai:')
]
# Create display names without the prefix
OPENAI_MODELS_DISPLAY = [
k.replace('openai:', '')
for k in OPENAI_MODELS_FULL
]
# Create and launch the interface using get_app utility
demo = get_app(
models=OPENAI_MODELS_FULL, # Use the full names with prefix
default_model=OPENAI_MODELS_FULL[5],
dropdown_label="Select OpenAI Model",
choices=OPENAI_MODELS_DISPLAY, # Display names without prefix
fill_height=True,
coder=True,
)
if __name__ == "__main__":
demo.launch()
|