Hugging Face's logo

Spaces:
asigalov61
/
Allegro-Music-Transformer
Running on Zero

App Files Community
1
asigalov61 commited on
Commit
b508df8
•
1 Parent(s): f47f09d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ import TMIDIX
16
 
17
  in_space = os.getenv("SYSTEM") == "spaces"
18
 
19
- providers = ['CPUExecutionProvider']
20
 
21
  #=================================================================================================
22
 
@@ -152,7 +152,7 @@ if __name__ == "__main__":
152
  opt = parser.parse_args()
153
 
154
 
155
- providers = ['CPUExecutionProvider']
156
 
157
  session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=providers)
158
 
 
16
 
17
  in_space = os.getenv("SYSTEM") == "spaces"
18
 
19
+ providers = ['CUDAExecutionProvider']
20
 
21
  #=================================================================================================
22
 
 
152
  opt = parser.parse_args()
153
 
154
 
155
+ providers = ['CUDAExecutionProvider']
156
 
157
  session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=providers)
158