Spaces:

ValdeciRodrigues
/

python-bot-ia

Runtime error

ValdeciRodrigues commited on 7 days ago

Commit

d474cef

verified ·

1 Parent(s): c65b0cc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 # Carregando o modelo StarCoder (especialista em programação)
 tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase")
-model = AutoModelForCausalLM.from_pretrained("bigcode/starcoderbase", trust_remote_code=True)
 gen = pipeline("text-generation", model=model, tokenizer=tokenizer)
 def process_input(user_input, files):

 # Carregando o modelo StarCoder (especialista em programação)
 tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase")
+model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder", trust_remote_code=True)
 gen = pipeline("text-generation", model=model, tokenizer=tokenizer)
 def process_input(user_input, files):