Update app.py

app.py

@@ -512,7 +512,7 @@ def main(input_text, max_new_tokens):
     encoded = torch.tensor(tokenizer.encode(input_text.strip())).unsqueeze(0).to(device)
 
     with torch.no_grad():
-        token_ids = generate_text_simple(
+        token_ids = generate(
             model=model, idx=encoded,
             max_new_tokens=max_new_tokens, context_size=context_size,
             top_k=25, temperature=1.4, text_to_token_ids=text_to_token_ids, tokenizer=tokenizer