asigalov61 commited on
Commit
5ffa949
1 Parent(s): 0d875dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from pytz import timezone
6
 
7
  import torch
8
 
9
- #import spaces
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
- # @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)))
 
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)))