Update app.py

app.py

@@ -17,9 +17,11 @@ def load_model():
     return model
 
 def generate_music_tensors(descriptions, duration: int):
-    # model = load_model()
-    model = load_model().to('cpu')
+    model = load_model()
 
+    # Manually set the device to CPU
+    device = torch.device('cpu')
+    model = model.to(device)
 
     model.set_generation_params(
         use_sampling=True,
@@ -65,7 +67,7 @@ st.set_page_config(
 
 def main():
     with st.sidebar:
-        st.header("""⚙️Generate Music ⚙️""",divider="rainbow")
+        st.header("""⚙️Generate Music ⚙️""", divider="rainbow")
         st.text("")
         st.subheader("1. Enter your music description.......")
         bpm = st.number_input("Enter Speed in BPM", min_value=60)
@@ -77,12 +79,12 @@ def main():
         
         st.subheader("2. Select time duration (In Seconds)")
         # time_slider = st.slider("Select time duration (In Seconds)", 0, 60, 10)
-        time_slider = st.slider("Select time duration (In Minutes)", 0,300,10, step=1)
+        time_slider = st.slider("Select time duration (In Minutes)", 0, 300, 10, step=1)
 
 
     st.title("""🎵 Song Lab AI 🎵""")
     st.text('')
-    left_co,right_co = st.columns(2)
+    left_co, right_co = st.columns(2)
     left_co.write("""Music Generation through a prompt""")
     left_co.write(("""PS : First generation may take some time ......."""))
     
@@ -117,4 +119,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-