Update app.py

app.py

@@ -6,9 +6,10 @@ first = """informal english: corn fields are all across illinois, visible once y
 
 @st.cache(allow_output_mutation=True)
 def get_model():
-    #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln2")
-    model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points3")
+    #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
+    #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points3")
+    model = AutoModelWithLMHead.from_pretrained("BigSalmon/InformalToFormalLincoln28")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
     return model, tokenizer