ansfarooq7 commited on
Commit
5d814e9
1 Parent(s): 93403e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,8 +11,7 @@ roberta_tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
11
  roberta_model = RobertaForMaskedLM.from_pretrained('roberta-base')
12
 
13
  gpt2_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
14
- gpt2_model = aitextgen(model="gpt2-medium")
15
- #gpt2_model = aitextgen()
16
 
17
  frequent_words = set()
18
 
 
11
  roberta_model = RobertaForMaskedLM.from_pretrained('roberta-base')
12
 
13
  gpt2_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
14
+ gpt2_model = aitextgen()
 
15
 
16
  frequent_words = set()
17