Update app.py

app.py

@@ -65,7 +65,8 @@ st.markdown(description, unsafe_allow_html=True)
 st.write('------')
 
 
-
+english_model = whisper.load_model("base.en")
+persian_model = Model.load("hezarai/whisper-small-fa")
 
 def cosine_sim(text1, text2):
     vectorizer = TfidfVectorizer().fit_transform([text1, text2])
@@ -81,11 +82,9 @@ def take_challenge(music_file, typed_lyrics, key, language):
             f.write(audio_value.getbuffer())
             
         if language == "en":
-            model = whisper.load_model("base.en")
-            user_lyrics = model.transcribe("user_sing.mp3", language=language)["text"]
+            user_lyrics = english_model.transcribe("user_sing.mp3", language=language)["text"]
         else:
-            model = Model.load("hezarai/whisper-small-fa")
-            user_lyrics = model.predict("user_sing.mp3")[0]["text"]
+            user_lyrics = persian_model.predict("user_sing.mp3")[0]["text"]
             
         st.write(user_lyrics)
         similarity_score = cosine_sim(typed_lyrics, user_lyrics)