Ultimate-Drums-Transformer

Running on Zero

asigalov61 commited on Sep 13, 2023

Commit

39106e3

1 Parent(s): 7654a54

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -155,7 +155,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', 'CPUExecutionProvider'])
     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'])
     print('Done!')
     app = gr.Blocks()