Update app.py

app.py

@@ -22,9 +22,6 @@ def generate_text(prompt, temperature, top_p):
     generated_text = prompt_with_eos  # Start with the initial prompt plus "EOS"
     prompt_length = len(generated_text)
 
-    if input_tokens.size(1) > 512:
-        generated_text = "ERROR, CONTEXT SIZE EXCEEDED"
-
     for _ in range(80):  # Adjust the range to control the number of tokens generated
         with torch.no_grad():
             outputs = model(input_tokens)