Update app.py

app.py

@@ -11,7 +11,7 @@ tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
 original_model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
 fine_tuned_model = GPT2LMHeadModel.from_pretrained('zmbfeng/FineTune-1')
 
-def create_response_original(input_str, input_num):
+def create_response_original(input_str, num_beams, input_num):
   print ("input_num="+str(num))
   #output_raw= generator(input_str)
   """print (output_raw)"""
@@ -33,7 +33,11 @@ def create_response_fine_tuned(input_str):
   #output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
   output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
   return (output_str)
-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")
+interface1 = gr.Interface(fn=create_response_original, inputs=[
+    gr.Textbox(label="input text here", lines=3),
+    gr.Number(label="num_beams (integer) explores the specified number of possible outputs and selects the most " +
+                "likely ones (specified in num_beams)", value=7),
+    gr.inputs.Number(label="Enter a whole number")], outputs="text", title="Original")
 interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
 demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
 # with gr.Blocks() as demo: