Update README.md

README.md

@@ -26,34 +26,8 @@ It is a full finetune for one epoch.
 * Original model [**Here**](https://huggingface.co/timpal0l/Mistral-7B-v0.1-flashback-v2)
 
 ## How to use:
-```python
-import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
-model_name = "timpal0l/Mistral-7B-v0.1-flashback-v2"
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
-
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
-model.eval()
-model.to(device)
-
-prompt = "Idag är det den bästa"
-input_ids = tokenizer(prompt, return_tensors="pt")["input_ids"].to(device)
-
-generated_token_ids = model.generate(
-    inputs=input_ids,
-    max_new_tokens=256,
-    do_sample=True,
-    temperature=0.8,
-    top_p=1,
-)[0]
-
-generated_text = tokenizer.decode(generated_token_ids)
-generated_text
-```
-```
-<s> Idag är det den bästa dagen i hela veckan, för nu tar det slut!\n\n>! Gnällfesten!\n\nJag sitter här, oerhört förvirrad, och försöker förstå varför vi ens måste fortsätta att existera efter döden. Jag menar, jag förstår ju egentligen att det aldrig kan ta slut, eller inte "ta slut" i den bemärkelsen att materian försvinner, men det är inte det jag pratar om.\n\nDöden, det faktum att man dör och aldrig kan uppleva livet igen. Det som är liv och ger livet en mening, det försvinner i döden. Och sen börjas det om, om och om igen. Varför behöver vi så många liv? Vi är ju inte ens medvetna av att vi någonsin har levt, så varför ska vi komma hit och bli medvetna hela tiden?\n\nDet här är en sådan fråga som jag aldrig kan få
+```bash
+ollama run timpal0l/mistral-swedish-flashback "Hejsan!"
 ```
 ## Data Format:
 To mimic the data format used in pre-training it has the following structure: