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])
|