Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 language:
-- en
+- hu
 license: apache-2.0
 tags:
 - text-generation-inference
@@ -9,14 +9,34 @@ tags:
 - mistral
 - trl
 base_model: unsloth/mistral-7b-bnb-4bit
+datasets:
+- SZTAKI-HLT/HunSum-1
 ---
 
-# Uploaded  model
+# Mistral-7b-0.1-hu
 
 - **Developed by:** Bazsalanszky
 - **License:** apache-2.0
 - **Finetuned from model :** unsloth/mistral-7b-bnb-4bit
 
-This mistral model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
+Ez a mistral 7b model magyar szövegre lett tanítva 10 000 véletlenszerűen kiválasztott cikken. Így valamivel szebben ír magyarul, mint az alap model. 
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+## Fontos
+
+Ez a modell NEM lett instrukciókra tanítva, valószínűleg nem fogja azokat követni.
+
+## Példa használat
+
+```python
+# Load model directly
+from transformers import AutoTokenizer, AutoModel
+
+tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
+model = AutoModel.from_pretrained("Bazsalanszky/Mistral-7b-0.1-hu")
+
+inputs = tokenizer("Magyarország\nFővárosa:", return_tensors = "pt").to("cpu")
+
+from transformers import TextStreamer
+text_streamer = TextStreamer(tokenizer)
+_ = model.generate(**inputs, streamer = text_streamer, max_new_tokens = 250)
+```