Update app.py

app.py

@@ -1,13 +1,24 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, PretrainedConfig
+
+
+pc = PretrainedConfig(
+    max_new_tokens=100,
+    num_beams=5,
+    early_stopping=True,
+    no_repeat_ngram_size=2,
+)
 
 pipe = pipeline(
-    "text-generation", model="RickMartel/GPT2_FT_By_NT_RAND_v5", device="cpu"
+    "text-generation", 
+    model="RickMartel/GPT2_FT_By_NT_RAND_v5", 
+    device="cpu",
+    config=pc
 )
 
 st.set_page_config(page_title="GPT2 4 Bible")
 
-txt = st.text_area('Enter prompt of a biblical nature. WARNING: results will not be exact.')
+txt = st.text_area('Enter prompt of a biblical nature. WARNING: Results may not be correct.')
 
 if txt and len(txt.strip()) > 0:
     txt = "<|startoftext|>" + txt