Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
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):
|