Update app.py

app.py

@@ -6,31 +6,43 @@ first = """informal english: corn fields are all across illinois, visible once y
 
 @st.cache(allow_output_mutation=True)
 def get_model():
-    print("hello")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
-    model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln2")
-    print("hello")
+    #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln2")
+    model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points3")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
-    print("hello")
     return model, tokenizer
     
 model, tokenizer = get_model()
 
 st.text('''How To Make Prompt:
-informal english: sega’s doing mobile gaming right with this approach to exploiting its back catalog, giving people a way to build up their library of classics on devices they’ll always have close to hand for cheap. it’s a far better experience than nintendo’s classic machine sales on its home gaming systems, which tend to be locked in and not passed between hardware generations.
-Translated into the Style of Abraham Lincoln: pursuing the proper course, sega's foray into mobile gaming entails leveraging its back catalog of classics on portable devices that ( are certain to be in reach whenever / will be at the ready for use at all times ) ( without spending a fortune / at minimal cost ). such an endeavor comes as a welcome alternative to nintendo's propensity to ( sell / offer / market ) home gaming systems that are rigidly bound to a single hardware generation.
+
+###
+
 Translated into the Style of Abraham Lincoln: at a time when nintendo has become inflexible, ( stubbornly bent on / firmly set on / unyielding in its insistence on / steadfastly transfixed by / uncompromising in its commitment to / rigidly decided on ) consoles that are tethered to a fixed iteration, sega diligently curates its legacy of classic video games on handheld devices.
 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 ( hasten their transition into the modern age / leap-frog them into the twenty-first century / facilitate their integration into contemporary life ).
-informal english: national parks are a big part of the us culture.
-Translated into the Style of Abraham Lincoln: the culture of the united states is ( inextricably ( bound up with / molded by / enriched by / enlivened by ) its ( serene / picturesque / pristine / breathtaking ) national parks ).
-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:''')
+informal english:
+
+###
+
+OR
+
+###
+- declining viewership facing the nba.
+- does not have to be this way.
+- in fact, many solutions exist.
+- the four point line would surely draw in eyes.
+text: failing to draw in the masses, the nba has ( fallen into / succumb to / bowed to ) disrepair. such does not have to be the case, however. in fact, a myriad of simple, relatively cheap ( solutions / interventions / enhancements ) could revive the league. the addition of the much-hyped four-point line would surely juice viewership.
+###
+-''')
 
 temp = st.sidebar.slider("Temperature", 0.7, 1.5)
 number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 50)
@@ -65,9 +77,11 @@ def run_generate(text, bad_words):
 with st.form(key='my_form'):
     text = st.text_area(label='Enter sentence', value=first)
     submit_button = st.form_submit_button(label='Submit')
+    submit_button2 = st.form_submit_button(label='Submit Log Probs')
     if submit_button:
       translated_text = run_generate(text, bad_words)
       st.write(translated_text if translated_text else "No translation found")
+    if submit_button2:
       with torch.no_grad():
         text2 = str(text)
         print(text2)