Update README.md
Browse files
README.md
CHANGED
|
@@ -30,7 +30,8 @@ pipe = pipeline(
|
|
| 30 |
device_map="auto"
|
| 31 |
)
|
| 32 |
text = """
|
| 33 |
-
Hur många ägg har jag? Jag hade 10 ägg, sen gav jag bort 5 ägg.
|
|
|
|
| 34 |
"""
|
| 35 |
generated = pipe(f"USER:{text}ASSISTANT:", max_length=512, temperature=0.6)
|
| 36 |
print(generated[0]["generated_text"].split("ASSISTANT: ")[1:][0])
|
|
|
|
| 30 |
device_map="auto"
|
| 31 |
)
|
| 32 |
text = """
|
| 33 |
+
Hur många ägg har jag? Jag hade 10 ägg, sen gav jag bort 5 ägg.
|
| 34 |
+
Sen fick jag 3 ägg av en kompis.
|
| 35 |
"""
|
| 36 |
generated = pipe(f"USER:{text}ASSISTANT:", max_length=512, temperature=0.6)
|
| 37 |
print(generated[0]["generated_text"].split("ASSISTANT: ")[1:][0])
|