birgermoell commited on
Commit
9ba59e7
·
verified ·
1 Parent(s): d3e86d8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +29 -0
README.md CHANGED
@@ -9,8 +9,37 @@ tags:
9
 
10
  ---
11
  # Eir
 
12
  <img src="eir.png" width="400px"/>
13
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  ## How to use
15
 
16
  ```bash
 
9
 
10
  ---
11
  # Eir
12
+ Eir är en svensk språkmodell fokuserar på hälsodomänen.
13
  <img src="eir.png" width="400px"/>
14
 
15
+ ```python
16
+ from transformers import AutoTokenizer
17
+ import transformers
18
+ import torch
19
+
20
+ model = "eir"
21
+ messages = [{"role": "user", "content": "Vad ska jag göra om jag är förkyld?"}]
22
+
23
+ tokenizer = AutoTokenizer.from_pretrained(model)
24
+ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
25
+ pipeline = transformers.pipeline(
26
+ "text-generation",
27
+ model=model,
28
+ torch_dtype=torch.float16,
29
+ device_map="auto",
30
+ )
31
+
32
+ outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.25, top_k=50, top_p=0.95)
33
+ print(outputs[0]["generated_text"])
34
+ ```
35
+
36
+ ```
37
+ Om du är förkyld är det viktigt att vila och dricka mycket vatten.
38
+ Om du har feber, kan du ta paracetamol eller ibuprofen.
39
+ Om du har svårare symptom som svår feber, svårare smärtor i halsen,
40
+ svårare svullnad i halsen, svårare svårigheter att andas, kan du behöva söka hjälp hos en läkare.
41
+ ```
42
+
43
  ## How to use
44
 
45
  ```bash