Spaces:

ppapenj
/

ysda_hf

Sleeping

ppapenj commited on Apr 7

Commit

94e1e0f

verified ·

1 Parent(s): 9507e10

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def load_dicts():
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract")
     model = AutoModelForSequenceClassification.from_pretrained(
-        "my_model/checkpoint-23000",
         num_labels=len(label2ind),
         problem_type="single_label_classification",
     )

 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract")
     model = AutoModelForSequenceClassification.from_pretrained(
+        "checkpoint-23000",
         num_labels=len(label2ind),
         problem_type="single_label_classification",
     )