update

utils/model.py

@@ -24,6 +24,7 @@ class Model():
             temperature=temp,
             num_return_sequences=1,
             eos_token_id=self.tokenizer.eos_token_id,
+            return_full_text=False
         )
         
         return sequences[-1]['generated_text'] #'\n'.join([seq['generated_text'] for seq in sequences])