Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
 model = AutoModelForCausalLM.from_pretrained("gpt2")
 
-def generate_response(input_text):
+def generate_response(message: str, history: list) -> str:
     inputs = tokenizer(input_text, return_tensors="pt")
     outputs = model.generate(inputs.input_ids, max_length=50)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)