SCM-LAB commited on
Commit
920eb70
·
verified ·
1 Parent(s): 6277d6e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -0
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ```python
2
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
+ import torch
4
+
5
+ model_name = "SCM-LAB/fluency-phobert-v2"
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
+ text = "Paris nằm ở chỗ nào?"
9
+ inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
10
+ outputs = model(**inputs)
11
+ predicted_probabilities = torch.softmax(outputs.logits, dim=1)
12
+ predicted_probabilities = predicted_probabilities.tolist()[0] # Chuyển tensor thành list
13
+ predicted_class = torch.argmax(outputs.logits, dim=1).item()
14
+
15
+ print("Nonfluency", predicted_probabilities[0])
16
+ print("Fluency" , predicted_probabilities[1])
17
+ ```