ValdeciRodrigues commited on
Commit
d474cef
·
verified ·
1 Parent(s): c65b0cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
6
 
7
  # Carregando o modelo StarCoder (especialista em programação)
8
  tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase")
9
- model = AutoModelForCausalLM.from_pretrained("bigcode/starcoderbase", trust_remote_code=True)
10
  gen = pipeline("text-generation", model=model, tokenizer=tokenizer)
11
 
12
  def process_input(user_input, files):
 
6
 
7
  # Carregando o modelo StarCoder (especialista em programação)
8
  tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase")
9
+ model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder", trust_remote_code=True)
10
  gen = pipeline("text-generation", model=model, tokenizer=tokenizer)
11
 
12
  def process_input(user_input, files):