Update README.md
Browse files
README.md
CHANGED
@@ -51,7 +51,7 @@ pip install -e .
|
|
51 |
```
|
52 |
|
53 |
## Load the model
|
54 |
-
```
|
55 |
# Load libraries
|
56 |
import torch
|
57 |
import torch.nn.functional as F
|
@@ -64,7 +64,7 @@ model.eval()
|
|
64 |
```
|
65 |
|
66 |
## Prediction
|
67 |
-
```
|
68 |
# Label List
|
69 |
emotion_label_list = [
|
70 |
'Anger',
|
|
|
51 |
```
|
52 |
|
53 |
## Load the model
|
54 |
+
```python
|
55 |
# Load libraries
|
56 |
import torch
|
57 |
import torch.nn.functional as F
|
|
|
64 |
```
|
65 |
|
66 |
## Prediction
|
67 |
+
```python
|
68 |
# Label List
|
69 |
emotion_label_list = [
|
70 |
'Anger',
|