Commit
•
19faf7a
1
Parent(s):
77d1363
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,7 +153,7 @@ if __name__ == "__main__":
|
|
| 153 |
opt = parser.parse_args()
|
| 154 |
|
| 155 |
print('Loading model...')
|
| 156 |
-
session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=['CUDAExecutionProvider'])
|
| 157 |
print('Done!')
|
| 158 |
|
| 159 |
app = gr.Blocks()
|
|
|
|
| 153 |
opt = parser.parse_args()
|
| 154 |
|
| 155 |
print('Loading model...')
|
| 156 |
+
session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
| 157 |
print('Done!')
|
| 158 |
|
| 159 |
app = gr.Blocks()
|