Spaces:
Running
Running
ElenaRyumina
commited on
Commit
β’
21cb4dd
1
Parent(s):
62621ad
Summary
Browse files- app/description.py +3 -1
- app/model.py +1 -1
app/description.py
CHANGED
@@ -17,4 +17,6 @@ Fearfully Angry, Fearfully Disgusted, Angrily Disgusted, Happily Disgusted).
|
|
17 |
|
18 |
<div class="app-flex-container">
|
19 |
<img src="https://img.shields.io/badge/version-v{config_data.APP_VERSION}-rc0" alt="Version">
|
20 |
-
|
|
|
|
|
|
17 |
|
18 |
<div class="app-flex-container">
|
19 |
<img src="https://img.shields.io/badge/version-v{config_data.APP_VERSION}-rc0" alt="Version">
|
20 |
+
<a href='https://github.com/ElenaRyumina/AVCER' target='_blank'><img src='https://img.shields.io/github/stars/ElenaRyumina/AVCER?style=flat' alt='GitHub' /></a>
|
21 |
+
</div>
|
22 |
+
"""
|
app/model.py
CHANGED
@@ -14,7 +14,7 @@ from app.config import config_data
|
|
14 |
from app.model_architectures import ResNet50, LSTMPyTorch, ExprModelV3
|
15 |
from transformers import AutoFeatureExtractor
|
16 |
|
17 |
-
device = "
|
18 |
|
19 |
def load_model(model_url, model_path):
|
20 |
try:
|
|
|
14 |
from app.model_architectures import ResNet50, LSTMPyTorch, ExprModelV3
|
15 |
from transformers import AutoFeatureExtractor
|
16 |
|
17 |
+
device = "cpu"
|
18 |
|
19 |
def load_model(model_url, model_path):
|
20 |
try:
|