zmbfeng commited on
Commit
cab4101
·
1 Parent(s): 71852c2

two interfaces using tab

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -32,10 +32,10 @@ def create_response_fine_tuned(input_str):
32
  #output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
33
  output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
34
  return (output_str)
35
- interface1 = gr.Interface(fn=create_response_original, inputs="text", outputs="text")
36
- interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text")
37
- with gr.Blocks() as demo:
38
- with gr.Row():
39
- interface1.show()
40
- interface2.show()
41
  demo.launch()
 
32
  #output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
33
  output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
34
  return (output_str)
35
+ interface1 = gr.Interface(fn=create_response_original, inputs="text", outputs="text", title="Original")
36
+ interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
37
+ demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
38
+ # with gr.Blocks() as demo:
39
+ # with gr.Row():
40
+ #
41
  demo.launch()