add input number

app.py

@@ -11,7 +11,8 @@ 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):
+def create_response_original(input_str, input_num):
+  print ("input_num="+str(num))
   #output_raw= generator(input_str)
   """print (output_raw)"""
 
@@ -32,7 +33,7 @@ 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="text", outputs="text", title="Original")
+interface1 = gr.Interface(fn=create_response_original, inputs=["text","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: