Duplicate from breadlicker45/music-gen

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: discord-gen
+emoji: 💻
+colorFrom: red
+colorTo: gray
+sdk: streamlit
+sdk_version: 1.19.0
+app_file: app.py
+pinned: false
+duplicated_from: breadlicker45/music-gen
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,44 @@
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+import xformers
+import streamlit as st
+import torch
+import time
+import random
+from transformers.trainer_utils import set_seed
+import time
+
+SEED = int(time.time())
+set_seed(SEED)
+
+#make columns
+col1, col2 = st.columns([2,1])
+
+#import model
+tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
+tokenizer.padding_side = 'left'
+model = AutoModelForCausalLM.from_pretrained("breadlicker45/gpt2-music")
+def get_model():
+    return pipeline('text-generation', model=model,
+ tokenizer=tokenizer, do_sample=True)
+
+
+#ui
+with col1:
+    prompt= st.text_input('input',
+        '''2623 2619 3970 3976 2607 3973 2735 3973 2598 3985 2726 3973 2607 4009 2735 3973 2598 3973 2726 3973 2607 3973 2735 4009''') 
+
+#gen text
+text = prompt
+
+generator = get_model()    
+gen = st.info('Generating text...')
+answer = generator(text, pad_token_id=tokenizer.eos_token_id, do_sample=True, max_length=350, min_length=80, temperature=0.7, top_k=2, num_beams=1, no_repeat_ngram_size=1, early_stopping=True)                      
+gen.empty()
+                       
+lst = answer[0]['generated_text']
+out = lst
+
+t = st.empty()
+for i in range(len(out)):
+    t.markdown("#### %s" % out[0:i])
+    time.sleep(0.04)

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+streamlit
+torch
+xformers