Update README.md

README.md

@@ -154,4 +154,9 @@ print(f"Model has {params} parameters")
 
 model.load_state_dict(torch.load("OmegLSTM.pt"))
 print(generate_text(model,char_to_idx,idx_to_char,"""Interests: \nYou: Hello\nStranger: """,gen_length=128,temperature=0.7,top_k=4))
-```
+```
+
+## Training
+The model was trained over 10 epochs on the entire dataset.
+Loss Graph:
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6316fb937b0ee0136e5f1220/7_oHKhzm4aS71tVUgD3B2.png)