Lps02 commited on
Commit
f1da251
·
verified ·
1 Parent(s): 0e5dff1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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]))