Alidr79 commited on
Commit
4fc9ca5
·
verified ·
1 Parent(s): 7f62a47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -10,7 +10,7 @@ import librosa
10
 
11
  dataset = load_dataset('SeyedAli/Persian-Speech-Dataset')
12
 
13
- dataset = dataset["test"].select(range(100))
14
 
15
 
16
  def set_seed(seed):
@@ -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_v2_best")
26
- model = AutoModelForTextToSpectrogram.from_pretrained("Alidr79/speecht5_v2_best")
27
 
28
 
29
  from speechbrain.inference.classifiers import EncoderClassifier
@@ -131,7 +131,7 @@ import gradio as gr
131
 
132
  slider = gr.Slider(
133
  minimum=0,
134
- maximum=100,
135
  value=86,
136
  step=1,
137
  label="Select a speaker"
 
10
 
11
  dataset = load_dataset('SeyedAli/Persian-Speech-Dataset')
12
 
13
+ dataset = dataset["test"]
14
 
15
 
16
  def set_seed(seed):
 
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
 
131
 
132
  slider = gr.Slider(
133
  minimum=0,
134
+ maximum=len(dataset["test"]),
135
  value=86,
136
  step=1,
137
  label="Select a speaker"