Amitontheweb commited on
Commit
00a32cf
1 Parent(s): b3565d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ token = os.environ.get("HF_TOKEN")
11
  # Load default model as GPT2 and other models
12
 
13
 
14
- tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
15
- model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
16
 
17
  tokenizer_gpt2 = AutoTokenizer.from_pretrained("openai-community/gpt2")
18
  model_gpt2 = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 
11
  # Load default model as GPT2 and other models
12
 
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
15
+ model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
16
 
17
  tokenizer_gpt2 = AutoTokenizer.from_pretrained("openai-community/gpt2")
18
  model_gpt2 = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")