Guided-Accompaniment-Transformer

Running on Zero

asigalov61 commited on Jan 22

Commit

19a2bd6

verified ·

1 Parent(s): bb2704c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -261,7 +261,7 @@ def Generate_Accompaniment(input_midi,
     print('=' * 70)
     print('Generating...')
-    model.to(DEVICE)
     model.eval()
     #==================================================================

     print('=' * 70)
     print('Generating...')
+    model.to(device_type)
     model.eval()
     #==================================================================