Spaces:

asigalov61
/

Allegro-Music-Transformer

Running on Zero

asigalov61 commited on Sep 13, 2023

Commit

5e1876e

•

1 Parent(s): 98256b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -157,7 +157,7 @@ if __name__ == "__main__":
     opt = parser.parse_args()
     print('Loading model...')
-    session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=['CUDAExecutionProvider'])
     print('Done!')
     app = gr.Blocks()

     opt = parser.parse_args()
     print('Loading model...')
+    session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
     print('Done!')
     app = gr.Blocks()