gorkemgoknar commited on
Commit
6b9d22e
1 Parent(s): f2fca0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ import random
3
  import torch
4
  from transformers import AutoConfig
5
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
 
6
 
7
  config = AutoConfig.from_pretrained('gorkemgoknar/gpt2chatbotenglish')
8
  model = GPT2LMHeadModel.from_pretrained('gorkemgoknar/gpt2chatbotenglish', config=config)
 
3
  import torch
4
  from transformers import AutoConfig
5
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
6
+ from itertools import chain
7
+
8
 
9
  config = AutoConfig.from_pretrained('gorkemgoknar/gpt2chatbotenglish')
10
  model = GPT2LMHeadModel.from_pretrained('gorkemgoknar/gpt2chatbotenglish', config=config)