Nick Canu commited on
Commit
a02dc94
1 Parent(s): ae88252
Files changed (1) hide show
  1. title_generator.py +1 -1
title_generator.py CHANGED
@@ -6,7 +6,7 @@ import spacy
6
  import torch
7
  from transformers import T5ForConditionalGeneration,T5Tokenizer
8
  import random
9
- import itemgetter
10
 
11
  #Custom text tokenizer from https://github.com/canunj/deconstructing_games by N Canu & K Chen
12
  def doc_text_preprocessing(ser):
 
6
  import torch
7
  from transformers import T5ForConditionalGeneration,T5Tokenizer
8
  import random
9
+ from operator import itemgetter
10
 
11
  #Custom text tokenizer from https://github.com/canunj/deconstructing_games by N Canu & K Chen
12
  def doc_text_preprocessing(ser):