Spaces:

BigSalmon
/

TextbookGPTInformalToFormal

Runtime error

BigSalmon commited on Jun 30, 2022

Commit

45a19ee

1 Parent(s): 4522a2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,8 +38,8 @@ def get_model():
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
     #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/TextbookInformalFormalEnglish")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/TextbookInformalFormalEnglish")
-    tokenizer = AutoTokenizer.from_pretrained("sberbank-ai/mGPT")
-    model = AutoModelForCausalLM.from_pretrained("sberbank-ai/mGPT")
     return model, tokenizer
 model, tokenizer = get_model()

     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
     #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/TextbookInformalFormalEnglish")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/TextbookInformalFormalEnglish")
+    tokenizer = AutoTokenizer.from_pretrained("gpt2-xl")
+    model = AutoModelForCausalLM.from_pretrained("gpt2-xl")
     return model, tokenizer
 model, tokenizer = get_model()