StringJammer commited on
Commit
22bad87
·
verified ·
1 Parent(s): b8f493b

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. see/app.py +1 -1
  2. see/inference.py +1 -1
  3. see/see_config.py +1 -1
see/app.py CHANGED
@@ -74,7 +74,7 @@ def get_model_status():
74
  return jsonify({
75
  'success': True,
76
  'loaded': classifier.is_loaded(),
77
- 'model_path': '/workspace/train_model/best_model'
78
  })
79
 
80
 
 
74
  return jsonify({
75
  'success': True,
76
  'loaded': classifier.is_loaded(),
77
+ 'model_path': '../best_model'
78
  })
79
 
80
 
see/inference.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  import torch
7
 
8
  # Model path
9
- MODEL_SAVE_PATH = '/workspace/train_model/best_model'
10
 
11
  # Emotion labels
12
  EMOTION_LABELS = [
 
6
  import torch
7
 
8
  # Model path
9
+ MODEL_SAVE_PATH = '../best_model'
10
 
11
  # Emotion labels
12
  EMOTION_LABELS = [
see/see_config.py CHANGED
@@ -5,7 +5,7 @@ Configuration File - Emotion Prediction Service
5
  import os
6
 
7
  # Model path - using trained model
8
- MODEL_SAVE_PATH = '/workspace/train_model/best_model'
9
 
10
  # Service port
11
  PORT = 8002
 
5
  import os
6
 
7
  # Model path - using trained model
8
+ MODEL_SAVE_PATH = '../best_model'
9
 
10
  # Service port
11
  PORT = 8002