Update README.md
Browse files
README.md
CHANGED
|
@@ -54,7 +54,7 @@ num epochs:10
|
|
| 54 |
## How to Get Started with the Model
|
| 55 |
Example usage:
|
| 56 |
|
| 57 |
-
from transformers import AutoImageProcessor, AutoModelForImageClassification, pipeline
|
| 58 |
|
| 59 |
pipe = pipeline("image-classification", model="HardlyHumans/Facial-expression-detection")
|
| 60 |
|
|
@@ -67,7 +67,7 @@ outputs = model(**inputs)
|
|
| 67 |
|
| 68 |
logits = outputs.logits
|
| 69 |
predicted_class_idx = logits.argmax(-1).item()
|
| 70 |
-
predicted_label = labels[predicted_class_idx]
|
| 71 |
|
| 72 |
|
| 73 |
|
|
|
|
| 54 |
## How to Get Started with the Model
|
| 55 |
Example usage:
|
| 56 |
|
| 57 |
+
'''from transformers import AutoImageProcessor, AutoModelForImageClassification, pipeline
|
| 58 |
|
| 59 |
pipe = pipeline("image-classification", model="HardlyHumans/Facial-expression-detection")
|
| 60 |
|
|
|
|
| 67 |
|
| 68 |
logits = outputs.logits
|
| 69 |
predicted_class_idx = logits.argmax(-1).item()
|
| 70 |
+
predicted_label = labels[predicted_class_idx]'''
|
| 71 |
|
| 72 |
|
| 73 |
|