arshadshk
/

notdiamond-4k-0001

Text Classification

Inference Endpoints

Model card Files Files and versions Community

arshadshk commited on Mar 24

Commit

16515a6

•

1 Parent(s): 3762b09

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -36,7 +36,8 @@ Inference:
     tokenizer = AutoTokenizer.from_pretrained("notdiamond/notdiamond-0001")
     model = AutoModelForSequenceClassification.from_pretrained("notdiamond/notdiamond-0001")
-    inputs = tokenizer(formatted_prompt, truncation=True, max_length=4096, return_tensors="pt")
     logits = model(**inputs).logits
     model_id = logits.argmax().item()

     tokenizer = AutoTokenizer.from_pretrained("notdiamond/notdiamond-0001")
     model = AutoModelForSequenceClassification.from_pretrained("notdiamond/notdiamond-0001")
+    inputs = tokenizer(formatted_prompt,
+      truncation=True, max_length=4096, return_tensors="pt")
     logits = model(**inputs).logits
     model_id = logits.argmax().item()