Update README.md
Browse files
README.md
CHANGED
@@ -3,6 +3,10 @@ license: mit
|
|
3 |
tags:
|
4 |
- text-generation-inference
|
5 |
pipeline_tag: text-generation
|
|
|
|
|
|
|
|
|
6 |
---
|
7 |
# Alice - Personal Assistant Fine-Tuned
|
8 |
|
@@ -19,4 +23,4 @@ model = AutoModelForCausalLM.from_pretrained("LPs02/Alice-model")
|
|
19 |
|
20 |
inputs = tokenizer("Leandro: Oi Alice, tudo bem?\nAlice:", return_tensors="pt")
|
21 |
outputs = model.generate(**inputs, max_new_tokens=50)
|
22 |
-
print(tokenizer.decode(outputs[0]))
|
|
|
3 |
tags:
|
4 |
- text-generation-inference
|
5 |
pipeline_tag: text-generation
|
6 |
+
language:
|
7 |
+
- pt
|
8 |
+
base_model:
|
9 |
+
- openai-community/gpt2-large
|
10 |
---
|
11 |
# Alice - Personal Assistant Fine-Tuned
|
12 |
|
|
|
23 |
|
24 |
inputs = tokenizer("Leandro: Oi Alice, tudo bem?\nAlice:", return_tensors="pt")
|
25 |
outputs = model.generate(**inputs, max_new_tokens=50)
|
26 |
+
print(tokenizer.decode(outputs[0]))
|