Update README.md

README.md

@@ -51,10 +51,27 @@ outputs = model.generate(
 for i in range(4):
     print(tokenizer.decode(outputs[i], skip_special_tokens=True, clean_up_tokenization_spaces=True))
 
-# The mechanical engineering field requires an understanding of core areas including mechanics, dynamics, thermodynamics, materials science, structural analysis, and electricity.
-# > Cheapity3 continuing:
+# INPUT: The mechanical engineering field requires an understanding of core areas including mechanics, dynamics, thermodynamics, materials science, structural analysis, and electricity.
+
+# > Cheapity3 continues with beam search:
+# ... The field of mechanical engineering is a broad field that includes many core areas of engineering.
+ 
+# > Cheapity3 continues with sampling and top_k=50:
 # ... Developing the knowledge base for these core areas will enable engineers to build their capabilities rapidly and efficiently. ...
 # ... The field of mechanics offers a variety and broad range for applications throughout the engineering/technological fields. ...
 # ... Mechanics generally is not understood by students. While they can be employed in the field, mechanical engineering ...
 # ... Introduction to mechanical engineering and core fields including chemical products, materials science, structural analysis, and geomatics ...
-``` 
+```
+
+## Pretty decent right?
+
+Hence, whenever you feel like GPT3 is too expensive, Cheapity3 comes to the rescue 🤗.
+
+## Model Training FYI
+- T5-base model
+- Trained on only 1M sentences from English, French and German text
+- Mostly text from wikipedia, arxiv and QA datasets
+- Learning rate: 0.00003
+- 2 epochs
+- Max input: 512 tokens
+- Max output: 128 tokens