Update app.py

app.py

@@ -5,10 +5,10 @@ import gradio as gr
 tokenizer = AutoTokenizer.from_pretrained('TrLOX/gpt2-tdk')
 model = AutoModelForCausalLM.from_pretrained('TrLOX/gpt2-tdk')
 
-def text_generation(input_text, seed):
-  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+def text_generation(keywords, domain, seed):
+  input_ids = tokenizer('keyword ' + keywords + ' domain ' + domain + ' title', return_tensors="pt").input_ids
   torch.manual_seed(seed) # Max value: 18446744073709551615
-  outputs = model.generate(input_ids, do_sample=True, min_length=50, max_length=200)
+  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
 
@@ -17,7 +17,7 @@ description = "Title and description generation by keywords"
 
 gr.Interface(
     text_generation,
-    [gr.inputs.Textbox(lines=2, label="Enter input text"), gr.inputs.Number(default=10, label="Enter seed number")],
+    [gr.inputs.Textbox(default='test 1,test 2',lines=2, label="Enter keywords"), gr.inputs.Textbox(lines=2, default='test.com',label="Enter domain"), gr.inputs.Number(default=10, label="Enter seed number")],
     [gr.outputs.Textbox(type="auto", label="Text Generated")],
     title=title,
     description=description,