Update app.py

app.py

@@ -13,7 +13,7 @@ model = GPT2LMHeadModel.from_pretrained('DarwinAnim8or/gpt-grug-1.5b', use_auth_
 # Define the function that generates text from a prompt
 def generate_text(prompt):
   input_ids = tokenizer.encode(prompt, return_tensors='pt')
-  output = model.generate(input_ids, max_length=80, do_sample=True)
+  output = model.generate(input_ids, max_new_tokens=80, do_sample=True)
   text = tokenizer.decode(output[0], skip_special_tokens=True)
   return text