Added samples

.gitignore

@@ -0,0 +1 @@
+*.ipynb

app.py

@@ -1,9 +1,12 @@
 from transformers import MusicgenForConditionalGeneration
 from transformers import AutoProcessor
 import scipy
-
 import streamlit as st
 
+# set seed
+import torch
+torch.manual_seed(2912)
+
 st.set_page_config(
     page_title="Plant Orchestra with GenAI",
     page_icon="🎵"
@@ -13,9 +16,7 @@ st.set_page_config(
 @st.cache_resource
 def initialise_model():
     try:
-        #processor = AutoProcessor.from_pretrained("musicgen-small")
         processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
-        # model = MusicgenForConditionalGeneration.from_pretrained("musicgen-small")
         model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
         return processor, model
     except Exception as e:
@@ -53,8 +54,6 @@ st.markdown("Generate music based on your own plant orchestra.")
 
 prompt = st.text_input(label='Prompt:', value='Sunflower, roses')
 if st.button("Generate Music"):
-    #with st.spinner("Initialising model..."):
-    #    processor, model = initialise_model()
     if processor is not None and model is not None:
         with st.spinner("Generating audio..."):
             results = generate_audio(processor, model, prompt)
@@ -64,25 +63,28 @@ if st.button("Generate Music"):
             st.audio(sample_rate=sampling_rate, data=results[0, 0].cpu().numpy(), format="audio/wav")
             
             
-# Add additional information and instructions for users
-st.sidebar.subheader("How to Use:")
-st.sidebar.write("1. Enter a plant condition prompt in the text input.")
-st.sidebar.write("2. Click the 'Generate Music' button to create music based on the provided prompt.")
+# Sidebar: How-to-use and Samples
+st.sidebar.header("How to Use:")
+st.sidebar.write("1. Enter a plant and condition (optional) in the text input. E.g. moss, 30C")
+st.sidebar.write("2. Click the 'Generate Music' button to create music based on the provided input.")
 st.sidebar.write("3. You can listen to the generated music and download it.")
 st.sidebar.write()
-st.sidebar.subheader('Samples 🎵')
-st.sidebar.write('prompt: Sunflower, roses')
-st.sidebar.audio('sound/sample.wav')
-st.sidebar.write("""prompt: Generate music for the Plant Orchestra based on the following plant conditions and temperatures:
-
-1. Rose plant: Temperature - 22°C
-2. Palm tree: Temperature - 28°C
-3. Orchid flower: Temperature - 20°C
-4. Succulent plant: Temperature - 25°C
-5. Maple tree: Temperature - 18°C
+st.sidebar.header('Samples 🎵')
+st.sidebar.write('Holland moss')
+st.sidebar.audio('sound/holland_moss.wav')
+st.sidebar.write('Nerve plant')
+st.sidebar.audio('sound/nerve_plant.wav')
+st.sidebar.write('Artillery plant')
+st.sidebar.audio('sound/artillery_plant.wav')
+st.sidebar.write('Malayan moss')
+st.sidebar.audio('sound/malayan_moss.wav')
+st.sidebar.write('Pilea')
+st.sidebar.audio('sound/pilea.wav')
+st.sidebar.write('Hydrocotyle tripartita')
+st.sidebar.audio('sound/hydrocotyle_tripartita.wav')
+st.sidebar.write('Oak leaf fig')
+st.sidebar.audio('sound/oak_leaf_fig.wav')
 
-Create a musical composition that represents the unique ambiance and vitality of each plant. Let the music reflect the varying temperatures and moods of these plants.""")
-st.sidebar.audio('sound/sample2.wav')
 
 # Footer
 st.write()