Merge branch 'main' of https://huggingface.co/spaces/JammyMachina/the-jam-machine-app into main

.gitignore

@@ -0,0 +1 @@
+__pycache__

README.md

@@ -1,6 +1,6 @@
 ---
 title: The Jam Machine
-emoji: ⚕
+emoji: 🎶
 colorFrom: darkblue
 colorTo: black
 sdk: gradio
@@ -10,14 +10,16 @@ app_file: playground.py
 pinned: true
 ---
 
-Contributors:
-
-Jean Simonnet:
-    https://github.com/misnaej
-    https://www.linkedin.com/in/jeansimonnet/
-Louis Demetz:
-    https://github.com/louis-demetz
-    https://www.linkedin.com/in/ldemetz/
-Halid Bayram:
-    https://github.com/m41w4r3exe
-    https://www.linkedin.com/in/halid-bayram-6b9ba861/
+[Presentation](pitch.com/public/417162a8-88b0-4472-a651-c66bb89428be)
+## Contributors:
+### Jean Simonnet:
+- [Github](https://github.com/misnaej)
+- [Linkedin](https://www.linkedin.com/in/jeansimonnet/)
+    
+### Louis Demetz:
+- [Github](https://github.com/louis-demetz)
+- [Linkedin](https://www.linkedin.com/in/ldemetz/)
+    
+### Halid Bayram:
+- [Github](https://github.com/m41w4r3exe)
+- [Linkedin](https://www.linkedin.com/in/halid-bayram-6b9ba861/)

load.py

@@ -1,6 +1,7 @@
 from transformers import GPT2LMHeadModel
 from transformers import PreTrainedTokenizerFast
 import os
+import torch
 
 
 class LoadModel:
@@ -31,6 +32,8 @@ class LoadModel:
         self.path = path
         self.device = device
         self.revision = revision
+        if torch.cuda.is_available():
+            self.device = "cuda"
 
     def load_model_and_tokenizer(self):
         model = self.load_model()

playground.py

@@ -123,7 +123,9 @@ def instrument_row(default_inst, row_id):
                 value=default_inst,
                 label="Instrument",
             )
-            temp = gr.Number(value=0.7, label="Creativity")
+            temp = gr.Dropdown(
+                [0.25, 0.5, 0.6, 0.7, 0.8], value=0.7, label="Creativity"
+            )
             density = gr.Dropdown([0, 1, 2, 3], value=3, label="Density")
 
         with gr.Column(scale=3):

requirements.txt

@@ -13,4 +13,5 @@ scipy
 datetime
 torch
 torchvision
-pyFluidSynth
+pyFluidSynth
+accelerate