Alidr79 commited on
Commit
d3d4abe
·
verified ·
1 Parent(s): 6e62053

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,8 +22,8 @@ set_seed(1)
22
  # Load model directly
23
  from transformers import AutoProcessor, AutoModelForTextToSpectrogram
24
 
25
- processor = AutoProcessor.from_pretrained("Alidr79/speecht5_movie_dialogue_30000_steps")
26
- model = AutoModelForTextToSpectrogram.from_pretrained("Alidr79/speecht5_movie_dialogue_30000_steps")
27
 
28
 
29
  from speechbrain.inference.classifiers import EncoderClassifier
 
22
  # Load model directly
23
  from transformers import AutoProcessor, AutoModelForTextToSpectrogram
24
 
25
+ processor = AutoProcessor.from_pretrained("Alidr79/speecht5_v2_best")
26
+ model = AutoModelForTextToSpectrogram.from_pretrained("Alidr79/speecht5_v2_best")
27
 
28
 
29
  from speechbrain.inference.classifiers import EncoderClassifier