Spaces:

asigalov61
/

Allegro-Music-Transformer

Running on Zero

asigalov61 commited on Sep 13, 2023

Commit

30685fa

•

1 Parent(s): b9dc011

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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