Update app.py

app.py

@@ -4,12 +4,15 @@ import time
 import gradio as gr
 
 # Load a pre-trained chatbot model
-chatbot = pipeline('text-generation', model='gpt2-large', max_length=500)
+chatbot = pipeline('text-generation', model='gpt2-large', max_length=200)
 
 def wrap(x):
   return textwrap.fill(x, replace_whitespace=False, fix_sentence_endings=True)
 
-def chat(prompt):
-  return chatbot(prompt)[0]['generated_text']
+def chat(input_text, additional_param):
+    generated_text = chatbot(input_text)[0]["generated_text"]
+    print(f"Generated text: {generated_text}")
+    print(f"Additional parameter: {additional_param}")
+    return generated_text
 
 gr.Interface(fn=chat, inputs="text", outputs="text").launch()