Update app.py

app.py

@@ -24,17 +24,11 @@ image_gradio_app = gr.Interface(
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
 chatbot_model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
-def predict_chatbot(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 = chatbot_model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id).tolist()
-    response = tokenizer.decode(history[0]).split("")
-
-    response_tuples = [(response[i], response[i+1]) for i in range(0, len(response)-1, 2)]
-    return response_tuples, history
+def echo(message, history):
+    return message
 
 chatbot_gradio_app = gr.ChatInterface(
-    fn=predict_chatbot,
+    fn=echo,
     title="Greta",
     theme=theme
 )
@@ -42,6 +36,6 @@ chatbot_gradio_app = gr.ChatInterface(
 # Combine both interfaces into a single app
 gr.TabbedInterface(
     [image_gradio_app, chatbot_gradio_app],
-    tab_names=["image","chatbot"],
+    tab_names=["Greta Image","Greta Chat"],
     theme=theme
 ).launch()