Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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[-1], | |
dropdown_label="Select OpenAI Model", | |
choices=OPENAI_MODELS_DISPLAY, # Display names without prefix | |
fill_height=True, | |
) | |
if __name__ == "__main__": | |
demo.launch() | |