bug fix

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import matplotlib.pyplot as plt
 from manipulate_model.utils import get_config_and_model, infere
 
 
@@ -10,14 +11,17 @@ def process(filepath):
     global manpulate_config
     out = infere(manipulate_model, filepath, manpulate_config)
     out = out.tolist()
-    return str(out)
+    plt.plot(out)
+    return str(out), plt
 demo = gr.Blocks()
 file_proc = gr.Interface(
     fn=process,
     inputs=[
         gr.Audio(sources=["microphone", "upload"], type="filepath", show_download_button=True, label="Speech file (<30s)", max_length=30),
     ],
-    outputs="text",
+    outputs=["text",
+             gr.Plot(label="Frame wise prediction")
+             ],
     title="Find the manipulation: Analyze 'Real' or 'Manipulated' audio.",
     description=(
         "Analyze, detect and localize manipulation in an audio with a click of a button. Upload a .wav or .flac file."

requirements.txt

@@ -2,4 +2,5 @@ gradio==3.50.2
 torch==2.4.0
 torchaudio==2.4.0
 numpy==1.24.4
-omegaconf==2.3.0
+omegaconf==2.3.0
+matplotlib