Print arguments

app.py

@@ -58,9 +58,9 @@ class Tango:
     def generate(self, prompt, steps = 100, guidance = 3, samples = 1, disable_progress = True):
         # Generate audio for a single prompt string
         with torch.no_grad():
-            print(self.model.parameters())
-            print(self.vae.parameters())
-            print(self.stft.parameters())
+            print(self.model.inference.__code__.co_varnames)
+            print(self.vae.decode_first_stage.__code__.co_varnames)
+            print(self.vae.decode_to_waveform.__code__.co_varnames)
             latents = self.model.inference([prompt], self.scheduler, steps, guidance, samples, disable_progress = disable_progress)
             mel = self.vae.decode_first_stage(latents)
             wave = self.vae.decode_to_waveform(mel)
@@ -107,9 +107,9 @@ def check(
 
 def update_output(output_format, output_number):
     return [
-        gr.update(visible = True),
-        gr.update(visible = (2 <= output_number)),
-        gr.update(visible = (output_number == 3)),
+        gr.update(format = output_format),
+        gr.update(format = output_format, visible = (2 <= output_number)),
+        gr.update(format = output_format, visible = (output_number == 3)),
         gr.update(visible = False)
     ]
 
@@ -164,7 +164,7 @@ with gr.Blocks() as interface:
         </ul>
         <br/>
         🐌 Slow process... ~2 hours. Your computer must <b><u>not</u></b> enter into standby mode.<br/>You can duplicate this space on a free account, it works on CPU.<br/>
-        <a href='https://huggingface.co/spaces/Fabrice-TIERCELIN/Text-to-Audio?duplicate=true'><img src='https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14'></a>
+        <a href='https://huggingface.co/spaces/Fabrice-TIERCELIN/Text-to-Audio?duplicate=true&hidden=public&hidden=public'><img src='https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14'></a>
         <br/>
         ⚖️ You can use, modify and share the generated sounds but not for commercial uses.
         """
@@ -180,9 +180,9 @@ with gr.Blocks() as interface:
 
     submit = gr.Button("🚀 Generate", variant = "primary")
 
-    output_audio_1 = gr.Video(label = "Generated Audio #1/3", autoplay = True)
-    output_audio_2 = gr.Video(label = "Generated Audio #2/3")
-    output_audio_3 = gr.Video(label = "Generated Audio #3/3")
+    output_audio_1 = gr.Audio(label = "Generated Audio #1/3", format = "wav", type="numpy", autoplay = True)
+    output_audio_2 = gr.Audio(label = "Generated Audio #2/3", format = "wav", type="numpy")
+    output_audio_3 = gr.Audio(label = "Generated Audio #3/3", format = "wav", type="numpy")
     information = gr.Label(label = "Information")
 
     submit.click(fn = update_seed, inputs = [