Matt commited on
Commit
21346d9
1 Parent(s): ea3b994

Update output

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,6 +29,7 @@ def apply_chat_template(template, test_conversation, add_generation_prompt, clea
29
  outputs = []
30
  conversation = json.loads(test_conversation)
31
  pr_snippet = (
 
32
  "tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)",
33
  f"tokenizer.chat_template = \"{template}\"",
34
  "tokenizer.push_to_hub(CHECKPOINT, create_pr=True)"
@@ -48,7 +49,7 @@ iface = gr.Interface(
48
  ],
49
  outputs=[
50
  gr.TextArea(label="Formatted conversation"),
51
- gr.TextArea(label="Code snippet to create PR", lines=3, show_copy_button=True)
52
  ]
53
  )
54
  iface.launch()
 
29
  outputs = []
30
  conversation = json.loads(test_conversation)
31
  pr_snippet = (
32
+ "CHECKPOINT = \"big-ai-company/cool-new-model\"\n"
33
  "tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)",
34
  f"tokenizer.chat_template = \"{template}\"",
35
  "tokenizer.push_to_hub(CHECKPOINT, create_pr=True)"
 
49
  ],
50
  outputs=[
51
  gr.TextArea(label="Formatted conversation"),
52
+ gr.TextArea(label="Code snippet to create PR", lines=3, show_label=True, show_copy_button=True)
53
  ]
54
  )
55
  iface.launch()