s3nh commited on
Commit
40ac173
1 Parent(s): 3817432

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -67,6 +67,12 @@ def inference(text, input):
67
  output = evaluate(instruction = text, input = input)
68
  return output
69
 
 
 
 
 
 
 
70
  io = gr.Interface(
71
  inference,
72
  inputs = [gr.Textbox(
 
67
  output = evaluate(instruction = text, input = input)
68
  return output
69
 
70
+ def model_tokenizer(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"]), "text"], "text")
75
+
76
  io = gr.Interface(
77
  inference,
78
  inputs = [gr.Textbox(