Update app.py

app.py

@@ -5,20 +5,12 @@ import gradio as gr
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
-def predict(message, history=[]):
-  new_user_input_ids = tokenizer.encode(message + tokenizer.eos_token, return_tensors='pt')
+def predict(input, history=[]):
+  new_user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors='pt')
   bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
   history = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id).tolist()
   response = tokenizer.decode(history[0]).replace("<|endoftext|>", "\n")
   
   return response, history
 
-iface = gr.Interface(
-    predict,
-    ["text", "state"],
-    ["chatbot", "state"],
-    allow_screenshot=False,
-    allow_flagging="never",
-    title="DialoGPT-large"
-)
-iface.launch()
+gr.Interface(fn=predict, title="DialoGPT-large", inputs=[gr.inputs.Textbox(placeholder="Write a text message as if writing a text message to a human."), "state"], outputs=["text", "state"]).launch()