Omaratef3221
/

flan-t5-base-dialogue-generator

Text2Text Generation

conversation-generator

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Omaratef3221 commited on Jan 6

Commit

4a4e331

•

1 Parent(s): 4b97baa

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,5 +52,5 @@ Dialogue:
 # Generate a response to an input statement
 input_ids = tokenizer(prompt, return_tensors='pt').input_ids
 output = model.generate(input_ids, top_p = 0.6, do_sample=True, temperature = 1.2, max_length = 512)
-print(tokenizer.decode(output[0], skip_special_tokens=True).replace('. ', '.\n'))
 ```

 # Generate a response to an input statement
 input_ids = tokenizer(prompt, return_tensors='pt').input_ids
 output = model.generate(input_ids, top_p = 0.6, do_sample=True, temperature = 1.2, max_length = 512)
+print(tokenizer.decode(output[0], skip_special_tokens=True).replace('#Person2#:', '\n#Person2#:').replace('#Person1#:', '\n#Person1#:'))
 ```