s3nh commited on
Commit
bef125f
1 Parent(s): 8548281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def choose_model(name):
71
 
72
  return load_model(name), load_tokenizer(name)
73
 
74
- name = gr.inputs.Dropdown(["s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish", "s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish"], "output")
75
  model = load_model(name)
76
  tokenizer = load_tokenizer(name)
77
 
 
71
 
72
  return load_model(name), load_tokenizer(name)
73
 
74
+ name = gr.inputs.Dropdown(["s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish", "s3nh/pythia-410m-91k-steps-self-instruct-polish"], label= "output")
75
  model = load_model(name)
76
  tokenizer = load_tokenizer(name)
77