Spaces:

asdc
/

temporal_expression_normalization

Runtime error

asdc commited on May 20

Commit

1719f05

verified ·

1 Parent(s): 0cf9d16

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ API_KEY = os.environ.get("ACESS_TOKEN")
 BASE_MODEL = "mistralai/Mistral-7B-v0.1"
 PEFT_ADAPTER = "asdc/Mistral-7B-multilingual-temporal-expression-normalization"
-tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
 base_model = AutoModelForCausalLM.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,

 BASE_MODEL = "mistralai/Mistral-7B-v0.1"
 PEFT_ADAPTER = "asdc/Mistral-7B-multilingual-temporal-expression-normalization"
+tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, token=API_KEY)
 base_model = AutoModelForCausalLM.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,