Update app.py

app.py

@@ -191,7 +191,7 @@ RESPUESTA: """
         with st.chat_message("assistant"):
             if 'GPT' not in genre:
                 if prompt.lower() == 'resume':
-                    llm_response = utils.summarise_doc(TRANSCRIPTION_PATH, model='llama')
+                    llm_response = utils.summarise_doc(TRANSCRIPTION_PATH, model_name='llama', model=MODEL)
                     st.markdown(llm_response)
                 else:
                     llm_response = qa_chain(prompt)['result']
@@ -214,7 +214,7 @@ RESPUESTA: """
                                 st_player(youtube_video_url.replace("?enablejsapi=1", "") + f'?start={start_time_seconds}&end={end_time_seconds}')
             else:
                 if prompt.lower() == 'resume':
-                    llm_response = utils.summarise_doc(TRANSCRIPTION_PATH, model='gpt')
+                    llm_response = utils.summarise_doc(TRANSCRIPTION_PATH, model_name='gpt')
                     st.markdown(llm_response)
                 else:
                     llm_response = utils.get_gpt_response(TRANSCRIPTION_PATH, prompt, logger)