Update app.py

app.py

@@ -128,7 +128,7 @@ def infer(ori_prompt):
 
     with torch.no_grad():
         with model.ema_scope():
-                wav_name = f'{prompt.strip().replace(" ", "-")}'
+                wav_name = f'{prompt["ori_caption"].strip().replace(" ", "-")}'
                 generator.gen_test_sample(prompt,wav_name=wav_name)
 
     print(f"Your samples are ready and waiting four you here: \nresults/test \nEnjoy.")