marc-es
/

orga-dynamic-1

sequence-classification

end-of-utterance

Model card Files Files and versions

marc-es commited on May 6

Commit

6e01b1d

·

verified ·

1 Parent(s): dc0e7c0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -59,7 +59,7 @@ from peft import PeftModel
 base = AutoModelForSequenceClassification.from_pretrained(
           "HuggingFaceTB/SmolLM2-135M-Instruct", num_labels=2)
 model = PeftModel.from_pretrained(base, "marc-es/orga-dynamic-1")
-tok   = AutoTokenizer.from_pretrained("latishab/turnsense")
 def is_end(text):
     out = model(**tok(text, return_tensors="pt"))[0]

 base = AutoModelForSequenceClassification.from_pretrained(
           "HuggingFaceTB/SmolLM2-135M-Instruct", num_labels=2)
 model = PeftModel.from_pretrained(base, "marc-es/orga-dynamic-1")
+tok   = AutoTokenizer.from_pretrained("marc-es/orga-dynamic-1")
 def is_end(text):
     out = model(**tok(text, return_tensors="pt"))[0]