DevashishBhake commited on
Commit
c1aba8c
·
1 Parent(s): d99dbf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ def get_sos_status(transcription, key_phrase):
51
 
52
  def main(audio):
53
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
54
- SPT_MODEL = "/code/SPT_model"
55
- model_name_or_path = "/code/SER_model"
56
  config = AutoConfig.from_pretrained(model_name_or_path)
57
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
58
  sampling_rate = feature_extractor.sampling_rate
 
51
 
52
  def main(audio):
53
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
54
+ SPT_MODEL = "./SPT_model"
55
+ model_name_or_path = "./SER_model"
56
  config = AutoConfig.from_pretrained(model_name_or_path)
57
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
58
  sampling_rate = feature_extractor.sampling_rate