ElenaRyumina commited on
Commit
21cb4dd
β€’
1 Parent(s): 62621ad
Files changed (2) hide show
  1. app/description.py +3 -1
  2. 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 = "cuda" if torch.cuda.is_available() else "cpu"
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: