Hugging Face's logo

Spaces:
stibiumghost
/
imitation_game
Runtime error

App Files Community
stibiumghost commited on
Commit
b5498b7
·
1 Parent(s): d3dc6c7

Update text_gen.py

Browse files
Files changed (1) hide show
  1. text_gen.py +1 -1
text_gen.py CHANGED
@@ -11,7 +11,7 @@ tokenizers = [transformers.AutoTokenizer.from_pretrained(model_names[0]),
11
 
12
  model = [transformers.AutoModelForSeq2SeqLM.from_pretrained(model_names[0]),
13
  transformers.BlenderbotForConditionalGeneration.from_pretrained(model_names[1]),
14
- transformers.GPT2Model.from_pretrained(model_names[2])]
15
 
16
 
17
  def generate_text(text, context, model_name, model, tokenizer, minimum=15, maximum=300):
 
11
 
12
  model = [transformers.AutoModelForSeq2SeqLM.from_pretrained(model_names[0]),
13
  transformers.BlenderbotForConditionalGeneration.from_pretrained(model_names[1]),
14
+ transformers.GPT2LMHeadModel.from_pretrained(model_names[2])]
15
 
16
 
17
  def generate_text(text, context, model_name, model, tokenizer, minimum=15, maximum=300):