valencar commited on
Commit
e250bea
·
1 Parent(s): 6b0f8d7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ model = AutoModelForCausalLM.from_pretrained("maritaca-ai/sabia-7b")
22
  prompt = "Question: Qual é o maior planeta do sistema solar ?"
23
  inputs = tokenizer(prompt, return_tensors="pt")
24
 
25
- # Generate
26
  generate_ids = model.generate(inputs.input_ids, max_length=100)
27
  output = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
28
 
 
22
  prompt = "Question: Qual é o maior planeta do sistema solar ?"
23
  inputs = tokenizer(prompt, return_tensors="pt")
24
 
25
+ # Generate
26
  generate_ids = model.generate(inputs.input_ids, max_length=100)
27
  output = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
28