Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from diffusers import AudioLDMControlNetPipeline, ControlNetModel
+import os
 from pretty_midi import PrettyMIDI
+import tempfile
 import torch
 import torchaudio
 
@@ -20,7 +22,8 @@ generator = torch.Generator(device)
 
 
 def predict(midi_file=None, prompt="", negative_prompt="", audio_length_in_s=5, random_seed=0, controlnet_conditioning_scale=1, num_inference_steps=20, guess_mode=False):
-    midi_file = midi_file.name
+    if not os.path.exists(midi_file):
+        midi_file = midi_file.name
     midi = PrettyMIDI(midi_file)
     audio = pipe(
         prompt,
@@ -56,7 +59,7 @@ with gr.Blocks(title="🎹 MIDI-AudioLDM", theme=gr.themes.Base(text_size=gr.the
         guess = gr.Checkbox(label="guess mode")
     btn = gr.Button("Generate")
     btn.click(predict, inputs=[midi, prompt, neg_prompt, duration, seed, cond, inf, guess], outputs=[audio])
-    gr.Examples(examples=[[gr.File("S00.mid"), "piano", "", 10, 25, 1.0, 20, False]], inputs=[midi, prompt, neg_prompt, duration, seed, cond, inf, guess], fn=predict, outputs=audio)
+    gr.Examples(examples=[["S00.mid", "piano", "", 10, 25, 1.0, 20, False]], inputs=[midi, prompt, neg_prompt, duration, seed, cond, inf, guess], fn=predict, outputs=audio)
     
 # demo = gr.Interface(
 #         fn=predict, inputs=[