asigalov61 commited on
Commit
23c274d
1 Parent(s): fc6d080

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -96,6 +96,8 @@ class JSMsgReceiver(gr.HTML):
96
  def get_block_name(self) -> str:
97
  return "html"
98
 
 
 
99
  if __name__ == "__main__":
100
 
101
  parser = argparse.ArgumentParser()
@@ -107,18 +109,17 @@ if __name__ == "__main__":
107
 
108
  providers = ['CPUExecutionProvider']
109
 
110
- session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=providers)
111
-
112
 
113
  app = gr.Blocks()
114
  with app:
115
- gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Midi Composer</h1>")
116
- gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.midi-composer&style=flat)\n\n"
117
- "Midi event transformer for music generation\n\n"
118
- "Demo for [SkyTNT/midi-model](https://github.com/SkyTNT/midi-model)\n\n"
119
  "[Open In Colab]"
120
- "(https://colab.research.google.com/github/SkyTNT/midi-model/blob/main/demo.ipynb)"
121
- " for faster running and longer generation"
122
  )
123
 
124
  js_msg = JSMsgReceiver()
 
96
  def get_block_name(self) -> str:
97
  return "html"
98
 
99
+ #=================================================================================================
100
+
101
  if __name__ == "__main__":
102
 
103
  parser = argparse.ArgumentParser()
 
109
 
110
  providers = ['CPUExecutionProvider']
111
 
112
+ # session = rt.InferenceSession('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.onnx', providers=providers)
 
113
 
114
  app = gr.Blocks()
115
  with app:
116
+ gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Allegro Music Transformer</h1>")
117
+ gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Allegro-Music-Transformer&style=flat)\n\n"
118
+ "Full-attention multi-instrumental music transformer featuring asymmetrical encoding with octo-velocity, and chords counters tokens, optimized for speed and performance\n\n"
119
+ "Check out [Allegro Music Transformer](https://github.com/asigalov61/Allegro-Music-Transformer) on GitHub!\n\n"
120
  "[Open In Colab]"
121
+ "(https://colab.research.google.com/github/asigalov61/Allegro-Music-Transformer/blob/main/Allegro_Music_Transformer_Composer.ipynb)"
122
+ " for faster execution and endless generation"
123
  )
124
 
125
  js_msg = JSMsgReceiver()