wangjin2000 commited on
Commit
5db5ccd
·
verified ·
1 Parent(s): d3e3ae7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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",