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

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 = "D:\kaggle_practice\KJSCE_hack\SERModel\SPT_model"
55
- model_name_or_path = "D:\kaggle_practice\KJSCE_hack\SERModel\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 = "/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