Update app.py

app.py

@@ -11,7 +11,10 @@ def text_generation(keywords, domain):
   torch.manual_seed(random.randint(0,18446744073709551615))
   outputs = model.generate(input_ids, do_sample=True, min_length=50, max_length=250)
   generated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)
-  return generated_text[0]
+  title_description_arr = generated_text.split(' title ')[1].split('description')
+  title = title_description_arr[0].strip()
+  description = title_description_arr[1].strip()
+  return title + "\r\n" + description
 
 title = "TDK GPT2"
 description = "Title and description generation by keywords"