s3nh commited on
Commit
8548281
·
1 Parent(s): 148548f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -71,7 +71,9 @@ def choose_model(name):
71
 
72
  return load_model(name), load_tokenizer(name)
73
 
74
- model, tokenizer = gr.Interface(choose_model, [gr.inputs.Dropdown(["s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish", "s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish"])])
 
 
75
 
76
  io = gr.Interface(
77
  inference,
 
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
 
78
  io = gr.Interface(
79
  inference,