Update app.py

app.py

@@ -18,6 +18,15 @@ from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVL
 import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
 import os
 
+with st.spinner("Loading the model"):
+    model_name = "Shaleen123/mistrallite_medical_qa"
+    
+    config = PeftConfig.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    model = PeftModel.from_pretrained(model, model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+
 
 # PDF 문서로부터 텍스트를 추출하는 함수입니다.
 def get_pdf_text(pdf_docs):
@@ -65,14 +74,6 @@ def get_vectorstore(text_chunks):
     return vectorstore # 생성된 벡터 스토어를 반환합니다.
 
 def get_conversation_chain(vectorstore):
-    # Replace 'microsoft/DialoGPT-large' with the desired model name    
-    model_name = "Shaleen123/mistrallite_medical_qa"
-    
-    config = PeftConfig.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    model = PeftModel.from_pretrained(model, model_name)
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-
     
     # 대화 기록을 저장하기 위한 메모리를 생성합니다.
     memory = ConversationBufferMemory(