|
import fn |
|
import gradio as gr |
|
from refresh import create_refresh_button |
|
|
|
with gr.Blocks() as demo: |
|
title = gr.Markdown('# llama-cpp-python server manager') |
|
info = gr.Markdown() |
|
url = gr.Textbox(label='URL', interactive=True) |
|
download_button = gr.Button(value='Download') |
|
model = gr.Dropdown(label='Model', choices=fn.list_model(), interactive=True) |
|
create_refresh_button(gr, model, lambda: None, lambda: {'choices': fn.list_model()}, 'refresh-button', interactive=True) |
|
set_model_button = gr.Button(value='Set Model') |
|
|
|
download_button.click( |
|
fn=fn.download, |
|
inputs=[url], |
|
outputs=[info], |
|
) |
|
|
|
set_model_button.click( |
|
fn=fn.set_model, |
|
inputs=[model], |
|
outputs=[], |
|
) |
|
|
|
if __name__ == '__main__': |
|
demo.launch() |
|
|