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",
|