Update gradio web server dir

tousika · Feb 3, 2024 · 5204cb4 · 5204cb4
1 parent 2c612f5
commit 5204cb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llava/serve/gradio_web_server.py b/llava/serve/gradio_web_server.py
@@ -314,7 +314,7 @@ def http_bot(state, model_selector, temperature, top_p, max_new_tokens, request:
 
 """
 
-def build_demo(embed_mode):
+def build_demo(embed_mode, cur_dir=None):
     textbox = gr.Textbox(show_label=False, placeholder="Enter text and press ENTER", container=False)
     with gr.Blocks(title="LLaVA", theme=gr.themes.Default(), css=block_css) as demo:
         state = gr.State()