Update app.py

app.py

@@ -22,7 +22,7 @@ phi2 = pipeline(
 )
 
 # Function that accepts a prompt and generates text using the phi2 pipeline
-def generate(message, chat_history, max_new_tokens):
+def generate(message, chat_history, max_new_tokens=21):
 
   instruction = "You are a helpful assistant to 'User'. You do not respond as 'User' or pretend to be 'User'. You only respond once as 'Assistant'."
   final_prompt = f"Instruction: {instruction}\n"