Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
aa942ba
1
Parent(s):
b508df8
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,8 +16,6 @@ import TMIDIX
|
|
| 16 |
|
| 17 |
in_space = os.getenv("SYSTEM") == "spaces"
|
| 18 |
|
| 19 |
-
providers = ['CUDAExecutionProvider']
|
| 20 |
-
|
| 21 |
#=================================================================================================
|
| 22 |
|
| 23 |
def generate(
|
|
@@ -151,10 +149,7 @@ if __name__ == "__main__":
|
|
| 151 |
parser.add_argument("--port", type=int, default=7860, help="gradio server port")
|
| 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 |
|
| 159 |
app = gr.Blocks()
|
| 160 |
with app:
|
|
|
|
| 16 |
|
| 17 |
in_space = os.getenv("SYSTEM") == "spaces"
|
| 18 |
|
|
|
|
|
|
|
| 19 |
#=================================================================================================
|
| 20 |
|
| 21 |
def generate(
|
|
|
|
| 149 |
parser.add_argument("--port", type=int, default=7860, help="gradio server port")
|
| 150 |
opt = parser.parse_args()
|
| 151 |
|
| 152 |
+
session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=['CUDAExecutionProvider'])
|
|
|
|
|
|
|
|
|
|
| 153 |
|
| 154 |
app = gr.Blocks()
|
| 155 |
with app:
|