Update app.py

app.py

@@ -20,7 +20,7 @@ processor = AutoProcessor.from_pretrained("suno/bark-small")
 model = BarkModel.from_pretrained("suno/bark-small").to(device)
 num_list = ["1","2","3","4","5","6","7","8","9","10"]
 lang_list = ["en","de"]
-
+#SAMPLE_RATE = 24_000
 def run_bark(text, n, lang):
     #history_prompt = []
     semantic_prompt=f"v2/{lang}_speaker_{int(n)-1}"
@@ -30,12 +30,16 @@ def run_bark(text, n, lang):
         voice_preset = semantic_prompt,
         return_tensors="pt",
     )
-    
-    speech_values = model.generate(**inputs, do_sample=True)
-    sampling_rate = model.generation_config.sample_rate
+    print("generating")
+    speech_values = model.generate(
+        **inputs, coarse_temperature = 0.8, temperature = 0.5, do_sample=True
+    )
+    #speech_values = model.generate(**inputs, do_sample=True)
+    #sampling_rate = model.generation_config.sample_rate
 
     #sampling_rate = model.config.sample_rate
-    #sampling_rate = 24000
+    sampling_rate = 24_000
+    print("writing")
     scipy.io.wavfile.write("bark_out.wav", rate=sampling_rate, data=speech_values.cpu().numpy().squeeze())
     return ("bark_out.wav")