zmbfeng commited on
Commit
c7eee8d
·
verified ·
1 Parent(s): 9afcb4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def create_response_fine_tuned(input_str):
33
  #output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
34
  output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
35
  return (output_str)
36
- interface1 = gr.Interface(fn=create_response_original, inputs=["text",gr.inputs.Number(label="Enter a whole number")], outputs="text", title="Original")
37
  interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
38
  demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
39
  # with gr.Blocks() as demo:
 
33
  #output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
34
  output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
35
  return (output_str)
36
+ interface1 = gr.Interface(fn=create_response_original, inputs=[gr.Textbox(label="input text here", lines=3),gr.inputs.Number(label="Enter a whole number")], outputs="text", title="Original")
37
  interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
38
  demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
39
  # with gr.Blocks() as demo: