Update app.py

app.py

@@ -14,8 +14,11 @@ def fn_clearvoice(aud):
     enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
     """
     output_wav_dict = myClearVoice(input_path='input.wav', online_write=False)
-    key = next(iter(output_wav_dict))
-    output_wav = output_wav_dict[key]
+    if isinstance(output_wav_dict, dict):
+        key = next(iter(output_wav_dict))
+        output_wav = output_wav_dict[key]
+    else:
+        output_wav = output_wav_dict
     torchaudio.save('enhanced.wav', output_wav.cpu(), 16000)
     return 'enhanced.wav'