Update app.py

app.py

@@ -20,22 +20,15 @@ def get_model():
     
 model, tokenizer = get_model()
 
-g = """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 ).
+g = """
+***
 
-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 ).
+original: sports teams are profitable for owners. [MASK], their valuations experience a dramatic uptick. 
+infill: sports teams are profitable for owners. ( accumulating vast sums / stockpiling treasure / realizing benefits / cashing in / registering robust financials / scoring on balance sheets ), their valuations experience a dramatic uptick. 
 
-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:"""
+original:"""
 
 with st.form(key='my_form'):
     prompt = st.text_area(label='Enter sentence', value=g)
@@ -50,7 +43,7 @@ with st.form(key='my_form'):
         logits, past_key_values = model(myinput, past_key_values = past_key_values, return_dict=False)
         logits = logits[0,-1]
         probabilities = torch.nn.functional.softmax(logits)
-        best_logits, best_indices = logits.topk(150)
+        best_logits, best_indices = logits.topk(250)
         best_words = [tokenizer.decode([idx.item()]) for idx in best_indices]
         text.append(best_indices[0].item())
         best_probabilities = probabilities[best_indices].tolist()