amongusrickroll68 commited on
Commit
cc2f9c1
1 Parent(s): 7960d4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,6 +3,9 @@ import tensorflow as tf
3
  import librosa
4
  git lfs install
5
  git clone https://huggingface.co/amongusrickroll68/MeloMind
 
 
 
6
  class TextToMusicGenerator:
7
  def __init__(self):
8
  self.model = tf.keras.models.load_model('path/to/model')
 
3
  import librosa
4
  git lfs install
5
  git clone https://huggingface.co/amongusrickroll68/MeloMind
6
+ from diffusers import DiffusionPipeline
7
+
8
+ pipeline = DiffusionPipeline.from_pretrained("amongusrickroll68/MeloMind")
9
  class TextToMusicGenerator:
10
  def __init__(self):
11
  self.model = tf.keras.models.load_model('path/to/model')