Update main.py

main.py

@@ -38,14 +38,16 @@ def convert_to_key():
     return f'{file_name}.musicxml', f'{file_name}_teaser.png'
 
 def predict(inp):
-    return inp
+    return convert_to_key()
 
 
 def run():
+    outputs = [gr.File(label='Download musicxml', type='file'), 
+               gr.File(label='Download detection image', type='file')]
     demo = gr.Interface(
         fn=predict,
         inputs='text',
-        outputs='text',
+        outputs=outputs,
         allow_flagging='never',
     )