Update app.py

app.py

@@ -11,7 +11,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 def generate_text():
     input_text = request.json['input_text']
     input_ids = tokenizer.encode(input_text, return_tensors='pt')
-    output_ids = model.generate(input_ids)
+    output_ids = model(**input_ids)
     output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
     return {'generated_text': output_text}