Update app.py

app.py

@@ -9,23 +9,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import tokenizers
 #os.environ["TOKENIZERS_PARALLELISM"] = "false"
 random.seed(None)
-first = """
-informal english: garage band has made people who know nothing about music good at creating music.
-Translated into the Style of Abraham Lincoln: garage band ( offers the uninitiated in music the ability to produce professional-quality compositions / catapults those for whom music is an uncharted art the ability the realize masterpieces / stimulates music novice's competency to yield sublime arrangements / begets individuals of rudimentary musical talent the proficiency to fashion elaborate suites ).
-
-informal english: chrome extensions can make doing regular tasks much easier to get done.
-Translated into the Style of Abraham Lincoln: chrome extensions ( yield the boon of time-saving convenience / ( expedite the ability to / unlock the means to more readily ) accomplish everyday tasks / turbocharges the velocity with which one can conduct their obligations ).
-
-informal english: broadband is finally expanding to rural areas, a great development that will thrust them into modern life.
-Translated into the Style of Abraham Lincoln: broadband is ( ( finally / at last / after years of delay ) arriving in remote locations / springing to life in far-flung outposts / inching into even the most backwater corners of the nation ) that will leap-frog them into the twenty-first century.
-
-informal english: google translate has made talking to people who do not share your language easier.
-Translated into the Style of Abraham Lincoln: google translate ( imparts communicability to individuals whose native tongue differs / mitigates the trials of communication across linguistic barriers / hastens the bridging of semantic boundaries / mollifies the complexity of multilingual communication / avails itself to the internationalization of discussion / flexes its muscles to abet intercultural conversation / calms the tides of linguistic divergence ).
-
-informal english: corn fields are all across illinois, visible once you leave chicago.
-Translated into the Style of Abraham Lincoln: corn fields ( permeate illinois / span the state of illinois / ( occupy / persist in ) all corners of illinois / line the horizon of illinois / envelop the landscape of illinois ), manifesting themselves visibly as one ventures beyond chicago.
-
-informal english:"""
+first = """informal english: corn fields are all across illinois, visible once you leave chicago.\nTranslated into the Style of Abraham Lincoln: corn fields ( permeate illinois / span the state of illinois / ( occupy / persist in ) all corners of illinois / line the horizon of illinois / envelop the landscape of illinois ), manifesting themselves visibly as one ventures beyond chicago.\ninformal english:"""
 suggested_text_list = [first]
 @st.cache(hash_funcs={tokenizers.Tokenizer: id, tokenizers.AddedToken: id})
 def load_model(model_name):
@@ -70,7 +54,7 @@ def extend(input_text, num_return_sequences, max_size=20, top_k=50, top_p=0.95):
         parsed_text = "שגיאה"
     return parsed_text
 if __name__ == "__main__":
-    st.title("Hebrew GPT Neo (Small)")
+    st.title("GPT2 Demo:")
     pre_model_path = "BigSalmon/InformalToFormalLincoln14"
     model, tokenizer = load_model(pre_model_path)
     stop_token = "<|endoftext|>"