Spaces:

kalinin-a-i
/

ML2_HW4_Kalinin

Running

Alexey Kalinin commited on Mar 31

Commit

84152d3

1 Parent(s): fd7668f

fix error

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -5,7 +5,7 @@ from joblib import load
 def load_model(path2chkpt: str, path2mapping: str):
-    model = AutoModelForSequenceClassification.from_pretrained("/home/jupyter/datasphere/project/hw4_nlp_ops/weights_20_classes/checkpoint-4500")
     tokenizer = AutoTokenizer.from_pretrained("distilbert/distilbert-base-cased")
     pipe = pipeline("text-classification",

 def load_model(path2chkpt: str, path2mapping: str):
+    model = AutoModelForSequenceClassification.from_pretrained(path2chkpt)
     tokenizer = AutoTokenizer.from_pretrained("distilbert/distilbert-base-cased")
     pipe = pipeline("text-classification",