chansung commited on
Commit
89c0f3c
1 Parent(s): c27096e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from vid2persona import init
4
  from vid2persona.pipeline import vlm
5
  from vid2persona.pipeline import llm
6
 
 
7
  init.auth_gcp()
8
  init.get_env_vars()
9
  prompt_tpl_path = "vid2persona/prompts"
@@ -96,7 +97,7 @@ with gr.Blocks(css="styles.css", theme=gr.themes.Soft()) as demo:
96
  user_input = gr.Textbox(placeholder="ask anything", interactive=False, elem_classes=["textbox-no-label", "textbox-no-top-bottom-borders"])
97
 
98
  with gr.Accordion("parameters' control pane", open=False):
99
- model_id = gr.Dropdown(choices=init.ALLOWED_LLM_FOR_HF_PRO_ACCOUNTS, value="HuggingFaceH4/zephyr-7b-beta", label="Model ID")
100
 
101
  with gr.Row():
102
  max_input_token_length = gr.Slider(minimum=1024, maximum=4096, value=4096, label="max-input-tokens")
 
4
  from vid2persona.pipeline import vlm
5
  from vid2persona.pipeline import llm
6
 
7
+ init.init_model("HuggingFaceH4/zephyr-7b-beta")
8
  init.auth_gcp()
9
  init.get_env_vars()
10
  prompt_tpl_path = "vid2persona/prompts"
 
97
  user_input = gr.Textbox(placeholder="ask anything", interactive=False, elem_classes=["textbox-no-label", "textbox-no-top-bottom-borders"])
98
 
99
  with gr.Accordion("parameters' control pane", open=False):
100
+ model_id = gr.Dropdown(choices=init.ALLOWED_LLM_FOR_HF_PRO_ACCOUNTS, value="HuggingFaceH4/zephyr-7b-beta", label="Model ID", visible=False)
101
 
102
  with gr.Row():
103
  max_input_token_length = gr.Slider(minimum=1024, maximum=4096, value=4096, label="max-input-tokens")