Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from diffusers import AudioLDMControlNetPipeline, ControlNetModel
 import os
 from pretty_midi import PrettyMIDI
-import tempfile
+from tempfile import _TemporaryFileWrapper
 import torch
 import torchaudio
 
@@ -22,7 +22,7 @@ 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):
- if not os.path.exists(midi_file):
+ if isinstance(midi_file, _TemporaryFileWrapper):
  midi_file = midi_file.name
  midi = PrettyMIDI(midi_file)
  audio = pipe(