Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -235,7 +235,8 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
|
|
| 235 |
eval_dataset=test_dataset,
|
| 236 |
tokenizer=tokenizer,
|
| 237 |
data_collator=DataCollatorForTokenClassification(tokenizer=tokenizer),
|
| 238 |
-
compute_metrics=compute_metrics
|
|
|
|
| 239 |
)
|
| 240 |
|
| 241 |
# Train and Save Model
|
|
@@ -248,6 +249,7 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
|
|
| 248 |
|
| 249 |
# Constants & Globals
|
| 250 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
|
|
| 251 |
|
| 252 |
MODEL_OPTIONS = [
|
| 253 |
"facebook/esm2_t6_8M_UR50D",
|
|
|
|
| 235 |
eval_dataset=test_dataset,
|
| 236 |
tokenizer=tokenizer,
|
| 237 |
data_collator=DataCollatorForTokenClassification(tokenizer=tokenizer),
|
| 238 |
+
compute_metrics=compute_metrics,
|
| 239 |
+
|
| 240 |
)
|
| 241 |
|
| 242 |
# Train and Save Model
|
|
|
|
| 249 |
|
| 250 |
# Constants & Globals
|
| 251 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 252 |
+
print("HF_TOKEN:",HF_TOKEN)
|
| 253 |
|
| 254 |
MODEL_OPTIONS = [
|
| 255 |
"facebook/esm2_t6_8M_UR50D",
|