Update app.py

app.py

@@ -11,7 +11,7 @@ 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)
-  title_description_arr = generated_text.split(' title ')[1].split('description')
+  title_description_arr = generated_text[0].split(' title ')[1].split('description')
   title = title_description_arr[0].strip()
   description = title_description_arr[1].strip()
   return title + "\r\n" + description