Create README.md

README.md

@@ -0,0 +1,43 @@
+---
+language:
+- en
+pipeline_tag: text2text-generation
+tags:
+- t5
+---
+
+```python
+from transformers import T5ForConditionalGeneration
+from transformers import T5TokenizerFast as T5Tokenizer
+model = "svjack/summary-dialogue-eng"
+device = "cpu"
+tokenizer = T5Tokenizer.from_pretrained(model)
+model = T5ForConditionalGeneration.from_pretrained(model).to(device).eval()
+
+prompt =  "The Wisconsin Territorial Centennial half dollar was designed by David Parsons and Benjamin Hawkins and minted by the United States Bureau of the Mint in 1936. The obverse (pictured) depicts a pick axe and lead ore, referring to the lead mining in early Wisconsin"
+prompt = "{}\nCandidates:Tom Jack".format(prompt)
+
+encode = tokenizer(prompt, return_tensors='pt').to(device)
+answer = model.generate(encode.input_ids,
+                       max_length = 128,
+    num_beams=2,
+    top_p = 0.95,
+    top_k = 50,
+    repetition_penalty = 2.5,
+    length_penalty=1.0,
+    early_stopping=True,
+                       )[0]
+decoded = tokenizer.decode(answer, skip_special_tokens=True)
+decoded.replace("Tom:", "\n").replace("Jack:", "\n").split("\n")
+```
+
+</br>
+
+```json
+['',
+ ' Have you seen the Wisconsin Territorial Centennial half dollar? ',
+ ' Yeah, it was designed by David Parsons and Benjamin Hawkins. ',
+ ' What is it? ',
+ " It's a half dollar with a pick axe and lead ore. ",
+ " That's great!"]
+```