aum27 commited on
Commit
ccb2e1d
·
verified ·
1 Parent(s): 5866a87

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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