Update app.py

app.py

@@ -39,7 +39,7 @@ def generate_text(prompt, temperature, top_p):
         
         decoded_token = tokenizer.decode(next_token.item())
         generated_text += decoded_token  # Append the new token to the generated text
-        if decoded_token == "#":  # Stop if the end of sequence token is generated
+        if decoded_token == "<":  # Stop if the end of sequence token is generated
             break
         yield generated_text[prompt_length:]  # Yield the generated text excluding the initial prompt plus "EOS"