Update text_gen.py

text_gen.py

@@ -15,8 +15,7 @@ model = [transformers.AutoModelForSeq2SeqLM.from_pretrained(model_names[0]),
 
 
 def generate_text(text, context, model_name, model, tokenizer, minimum=15, maximum=300):
-    if context:
-        text = f'{context} {text}'
+    text = f'{context} {text}'
     if 'GODEL' in model_name:
         text = f'Instruction: you need to response discreetly. [CONTEXT] {text}'
         text.replace('\t', ' EOS ')
@@ -34,4 +33,4 @@ def capitalization(line):
         line = f'{mark} '.join([part.strip()[0].upper() +  part.strip()[1:] for part in line.split(mark) if len(part) > 1])
     line = ' '.join([word.capitalize() if word.translate(str.maketrans('', '', string.punctuation)) == 'i'
                     else word for word in line.split()])
-    return line + end
+    return line.replace(' i\'',  ' I\'') + end