Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 import openai
 import requests
 from gtts import gTTS
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import MusicgenForCausalLM, MusicgenConfig, AutoTokenizer
 import os
 from io import BytesIO
 from pydub import AudioSegment
@@ -13,14 +13,16 @@ import tempfile
 
 # Configure API keys
 HUGGING_FACE_API_KEY = "voicebot"
-OPENAI_API_KEY = "your_openai_api_key"
+OPENAI_API_KEY = "Testing API"
 openai.api_key = OPENAI_API_KEY
 
 # Initialize the Hugging Face model and tokenizer
+@st.cache_resource  # Cache the model to avoid reloading on every run
 def load_model():
     model_name = "facebook/musicgen-small"
+    config = MusicgenConfig.from_pretrained(model_name, use_auth_token=HUGGING_FACE_API_KEY)
+    model = MusicgenForCausalLM.from_pretrained(model_name, config=config, use_auth_token=HUGGING_FACE_API_KEY)
     tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=HUGGING_FACE_API_KEY)
-    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HUGGING_FACE_API_KEY)
     return model, tokenizer
 
 model, tokenizer = load_model()
@@ -85,3 +87,5 @@ if audio_file is not None:
 else:
     st.write("Please upload an audio file to get started.")
 
+
+