yasuogg commited on
Commit
6f848f3
·
verified ·
1 Parent(s): 7efa79e

Replace example reference audio with 3 new refs (1/2/3.wav)

Browse files
.gitattributes CHANGED
@@ -34,3 +34,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  examples/ALG.wav filter=lfs diff=lfs merge=lfs -text
 
 
 
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  examples/ALG.wav filter=lfs diff=lfs merge=lfs -text
37
+ examples/1.wav filter=lfs diff=lfs merge=lfs -text
38
+ examples/2.wav filter=lfs diff=lfs merge=lfs -text
39
+ examples/3.wav filter=lfs diff=lfs merge=lfs -text
__pycache__/app.cpython-312.pyc DELETED
Binary file (9.66 kB)
 
app.py CHANGED
@@ -57,8 +57,23 @@ vocab_path = hf_hub_download(MODEL_REPO, "vocab.txt")
57
  vocoder = load_vocoder(vocoder_name="vocos", is_local=False, device="cpu")
58
  model = load_model(DiT, V1_CFG, ckpt_path, vocab_file=vocab_path, device="cpu")
59
 
60
- EXAMPLE_REF = os.path.join(os.path.dirname(__file__), "examples", "ALG.wav")
61
- EXAMPLE_REF_TEXT = "أنيا هكا باغية ناكل هكا أني ن نشوف فيها الحاجة هذيكا."
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
62
 
63
  # The model was trained on unvocalized Arabic script only: Latin letters and
64
  # digits fall outside its vocabulary and produce unpredictable audio.
@@ -188,11 +203,17 @@ with gr.Blocks(title="KasbahTTS V0", theme=gr.themes.Soft()) as demo:
188
 
189
  run.click(generate, inputs=inputs, outputs=outputs)
190
 
191
- if os.path.exists(EXAMPLE_REF):
 
 
 
 
 
 
192
  gr.Examples(
193
  examples=[
194
- [EXAMPLE_REF, EXAMPLE_REF_TEXT, "واش راك خويا، لاباس عليك؟ صباح الخير."],
195
- [EXAMPLE_REF, EXAMPLE_REF_TEXT, "من القصبة للعالم، هذا أول موديل يهدر بالدارجة الجزائرية."],
196
  ],
197
  inputs=[ref_audio, ref_text, gen_text],
198
  )
 
57
  vocoder = load_vocoder(vocoder_name="vocos", is_local=False, device="cpu")
58
  model = load_model(DiT, V1_CFG, ckpt_path, vocab_file=vocab_path, device="cpu")
59
 
60
+ EXAMPLES_DIR = os.path.join(os.path.dirname(__file__), "examples")
61
+ # (reference wav, exact transcript of that wav) transcripts must match the
62
+ # audio verbatim; all refs are kept under f5-tts's 12s reference clip limit.
63
+ EXAMPLE_REFS = [
64
+ (
65
+ os.path.join(EXAMPLES_DIR, "1.wav"),
66
+ "مَثَلاً ما تَلْڨاشْ واحدْ يقول لك شْجْرة قْلَقتني، شْجْرة مَرَضَتْلي حْيَاتي. تْلْڨاه غِيرْ يْڨُول لك هَذ الإنْسَان فْلَان وْ فْلَان وْ فْلَان. تسما مَشَاكِلْنا كَامَل مَنْ عَنْد النَّاس.",
67
+ ),
68
+ (
69
+ os.path.join(EXAMPLES_DIR, "2.wav"),
70
+ "الطالبات يروحوا ليها يتغداو، كنت أنا نروح بعد لي كور باش نتغدى، وكان كاين واحد الشاب دائماً يشوف فيا ويحب يحكي معايا، ويعاملني معاملة خاصة.",
71
+ ),
72
+ (
73
+ os.path.join(EXAMPLES_DIR, "3.wav"),
74
+ "الدراهم ولا القدره المعيشيه كيما الفلاح عمر هذا كان عادي معندوش و مخصوش عايش على قد حال",
75
+ ),
76
+ ]
77
 
78
  # The model was trained on unvocalized Arabic script only: Latin letters and
79
  # digits fall outside its vocabulary and produce unpredictable audio.
 
203
 
204
  run.click(generate, inputs=inputs, outputs=outputs)
205
 
206
+ available_refs = [(wav, text) for wav, text in EXAMPLE_REFS if os.path.exists(wav)]
207
+ if available_refs:
208
+ gen_samples = [
209
+ "واش راك خويا، لاباس عليك؟ صباح الخير.",
210
+ "من القصبة للعالم، هذا أول موديل يهدر بالدارجة الجزائرية.",
211
+ "اليوم الجو شباب، قلت نخرج نتمشى شوية في وسط البلاد.",
212
+ ]
213
  gr.Examples(
214
  examples=[
215
+ [wav, text, gen_samples[i % len(gen_samples)]]
216
+ for i, (wav, text) in enumerate(available_refs)
217
  ],
218
  inputs=[ref_audio, ref_text, gen_text],
219
  )
examples/{ALG.wav → 1.wav} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:187b5992f7b068371f2769860a8b776e50abcc4168d9d24811ebb97ec0b8664f
3
- size 294174
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:700c386039290655aa05e7d5dd734462550ee08be4955735b03842ac91b0533d
3
+ size 368492
examples/2.wav ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ae9185bbedbcdeaaab72ed40e843c333e4ae1ac47cb0e12de6e86eafc23db46
3
+ size 342444
examples/3.wav ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1362aa496c983f698235219910fec486d06f1a2e6610a9292e38e069cec55c65
3
+ size 610604