DevashishBhake commited on
Commit
6c75f9b
·
1 Parent(s): fbfe17a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,8 +58,8 @@ def get_sos_status(transcription, key_phrase):
58
 
59
  def main(audio):
60
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
61
- SPT_MODEL = "D:\kaggle_practice\KJSCE_hack\SERModel\SPT_model"
62
- model_name_or_path = "D:\kaggle_practice\KJSCE_hack\SERModel\SER_model"
63
  config = AutoConfig.from_pretrained(model_name_or_path)
64
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
65
  sampling_rate = feature_extractor.sampling_rate
 
58
 
59
  def main(audio):
60
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
61
+ SPT_MODEL = "./SPT_model"
62
+ model_name_or_path = "./SER_model"
63
  config = AutoConfig.from_pretrained(model_name_or_path)
64
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
65
  sampling_rate = feature_extractor.sampling_rate