Spaces:
Running
on
Zero
Running
on
Zero
asigalov61
commited on
Commit
•
5ffa949
1
Parent(s):
0d875dc
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from pytz import timezone
|
|
6 |
|
7 |
import torch
|
8 |
|
9 |
-
|
10 |
import gradio as gr
|
11 |
|
12 |
from x_transformer_1_23_2 import *
|
@@ -20,7 +20,7 @@ import matplotlib.pyplot as plt
|
|
20 |
|
21 |
# =================================================================================================
|
22 |
|
23 |
-
|
24 |
def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
|
25 |
print('=' * 70)
|
26 |
print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
|
|
|
6 |
|
7 |
import torch
|
8 |
|
9 |
+
import spaces
|
10 |
import gradio as gr
|
11 |
|
12 |
from x_transformer_1_23_2 import *
|
|
|
20 |
|
21 |
# =================================================================================================
|
22 |
|
23 |
+
@spaces.GPU
|
24 |
def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
|
25 |
print('=' * 70)
|
26 |
print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
|