Hugging Face's logo

amaai-lab
/
text2midi

PyTorch
music
text-to-music
symbolic-music
Model card Files Community
keshavbhandari commited on
Commit
19860b2
·
verified ·
1 Parent(s): 4cab6a4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -12,7 +12,7 @@ tags:
12
 
13
  # Text2midi: Generating Symbolic Music from Captions
14
 
15
- [Demo](https://huggingface.co/spaces/amaai-lab/text2midi) | [Model](https://huggingface.co/amaai-lab/text2midi) | [Website and Examples](https://github.com/AMAAI-Lab/text2midi) | [Paper](https://arxiv.org/abs/TBD) | [Dataset](https://huggingface.co/datasets/amaai-lab/MidiCaps)
16
 
17
  [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/amaai-lab/text2midi)
18
  </div>
@@ -30,6 +30,9 @@ tags:
30
  Generate symbolic music from a text prompt:
31
 
32
  ```python
 
 
 
33
  from transformers import T5Tokenizer
34
  from model.transformer_model import Transformer
35
  from huggingface_hub import hf_hub_download
 
12
 
13
  # Text2midi: Generating Symbolic Music from Captions
14
 
15
+ [Demo](https://huggingface.co/spaces/amaai-lab/text2midi) | [Model](https://huggingface.co/amaai-lab/text2midi) | [Website and Examples](https://github.com/AMAAI-Lab/text2midi) | [Paper](https://arxiv.org/abs/2412.16526) | [Dataset](https://huggingface.co/datasets/amaai-lab/MidiCaps)
16
 
17
  [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/amaai-lab/text2midi)
18
  </div>
 
30
  Generate symbolic music from a text prompt:
31
 
32
  ```python
33
+ import pickle
34
+ import torch
35
+ import torch.nn as nn
36
  from transformers import T5Tokenizer
37
  from model.transformer_model import Transformer
38
  from huggingface_hub import hf_hub_download