Update app.py

app.py

@@ -13,6 +13,7 @@ number_of_outputs = st.sidebar.slider("Number of Outputs", 50, 350)
 
 @st.cache(allow_output_mutation=True)
 def get_model():
+    #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo1.3BPointsLincolnFormalInformal")
     #model = #AutoModelWithLMHead.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln6")
     #model = AutoModelWithLMHead.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln5")
     #model = AutoModelWithLMHead.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln4")
@@ -34,7 +35,7 @@ def get_model():
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPT2Neo1.3BPoints")
-    model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo1.3BPointsLincolnFormalInformal")
+    model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
     return model, tokenizer
     
 model, tokenizer = get_model()