aum27 commited on
Commit
b684f27
·
verified ·
1 Parent(s): 736c913

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -6
README.md CHANGED
@@ -42,16 +42,13 @@ The model has been fine-tuned using the following hyperparameters:
42
 
43
 
44
 
45
- - **Developed by:** Hardly Humans club
46
- - **Model type:** Vision transformer
47
- - **License:** MIT
48
- - **Finetuned from model:**
49
 
50
 
51
  ## How to Get Started with the Model
52
  Example usage:
53
 
54
- '''python
55
  from transformers import AutoImageProcessor, AutoModelForImageClassification, pipeline
56
 
57
  pipe = pipeline("image-classification", model="HardlyHumans/Facial-expression-detection")
@@ -66,7 +63,7 @@ outputs = model(**inputs)
66
  logits = outputs.logits
67
  predicted_class_idx = logits.argmax(-1).item()
68
  predicted_label = labels[predicted_class_idx]
69
- '''
70
 
71
 
72
 
@@ -75,6 +72,12 @@ predicted_label = labels[predicted_class_idx]
75
  The net estimated CO2 emission using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) scale is around 8.82kg of CO2.
76
 
77
 
 
 
 
 
 
 
78
  - **Hardware Type:** T4
79
  - **Hours used:** 8+27
80
  - **Cloud Provider:** Google collabotary service
 
42
 
43
 
44
 
45
+
 
 
 
46
 
47
 
48
  ## How to Get Started with the Model
49
  Example usage:
50
 
51
+ ```python
52
  from transformers import AutoImageProcessor, AutoModelForImageClassification, pipeline
53
 
54
  pipe = pipeline("image-classification", model="HardlyHumans/Facial-expression-detection")
 
63
  logits = outputs.logits
64
  predicted_class_idx = logits.argmax(-1).item()
65
  predicted_label = labels[predicted_class_idx]
66
+ ```
67
 
68
 
69
 
 
72
  The net estimated CO2 emission using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) scale is around 8.82kg of CO2.
73
 
74
 
75
+
76
+ - **Developed by:** Hardly Humans club, IIT Dharwad
77
+ - **Model type:** Vision transformer
78
+ - **License:** MIT
79
+ - **Finetuned from model:** google/vit-base-patch16-224-in21k
80
+
81
  - **Hardware Type:** T4
82
  - **Hours used:** 8+27
83
  - **Cloud Provider:** Google collabotary service