Jofthomas HF staff commited on
Commit
f5e5ce2
1 Parent(s): 3fb14f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,6 +6,8 @@ from huggingface_hub import login
6
 
7
  HUGGINGFACEHUB_API_TOKEN = os.environ.get("HF_TOKEN")
8
 
 
 
9
  demo_conversation = """[
10
  {"role": "system", "content": "You are a helpful chatbot."},
11
  {"role": "user", "content": "Hi there!"},
@@ -53,7 +55,7 @@ def apply_chat_template(model_name, test_conversation, add_generation_prompt, cl
53
  return str(e)
54
 
55
  with gr.Blocks() as demo:
56
- model_name_input = gr.Textbox(label="Model Name", placeholder="Enter model name")
57
  template_dropdown = gr.Dropdown(label="Template Name", choices=[], interactive=True)
58
  conversation_input = gr.TextArea(value=demo_conversation, lines=6, label="Conversation")
59
  add_generation_prompt_checkbox = gr.Checkbox(value=False, label="Add generation prompt")
 
6
 
7
  HUGGINGFACEHUB_API_TOKEN = os.environ.get("HF_TOKEN")
8
 
9
+ default_model="meta-llama/Meta-Llama-3-8B-Instruct"
10
+
11
  demo_conversation = """[
12
  {"role": "system", "content": "You are a helpful chatbot."},
13
  {"role": "user", "content": "Hi there!"},
 
55
  return str(e)
56
 
57
  with gr.Blocks() as demo:
58
+ model_name_input = gr.Textbox(label="Model Name", placeholder="Enter model name", value=default_model)
59
  template_dropdown = gr.Dropdown(label="Template Name", choices=[], interactive=True)
60
  conversation_input = gr.TextArea(value=demo_conversation, lines=6, label="Conversation")
61
  add_generation_prompt_checkbox = gr.Checkbox(value=False, label="Add generation prompt")