elapt1c commited on
Commit
dc31ddc
·
verified ·
1 Parent(s): 6eae396

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
 
3
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
4
- model_name = 'elapt1c/ElapticAI-1a'
5
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
6
  model = GPT2LMHeadModel.from_pretrained(model_name)
7
 
 
1
  import streamlit as st
2
 
3
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
4
+ model_name = 'gpt2'
5
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
6
  model = GPT2LMHeadModel.from_pretrained(model_name)
7