Spaces:
Sleeping
Sleeping
Commit
Β·
4f913b7
1
Parent(s):
e24629e
Update app.py
Browse files
app.py
CHANGED
@@ -18,6 +18,15 @@ from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVL
|
|
18 |
import tempfile # μμ νμΌμ μμ±νκΈ° μν λΌμ΄λΈλ¬λ¦¬μ
λλ€.
|
19 |
import os
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
# PDF λ¬Έμλ‘λΆν° ν
μ€νΈλ₯Ό μΆμΆνλ ν¨μμ
λλ€.
|
23 |
def get_pdf_text(pdf_docs):
|
@@ -65,14 +74,6 @@ def get_vectorstore(text_chunks):
|
|
65 |
return vectorstore # μμ±λ λ²‘ν° μ€ν μ΄λ₯Ό λ°νν©λλ€.
|
66 |
|
67 |
def get_conversation_chain(vectorstore):
|
68 |
-
# Replace 'microsoft/DialoGPT-large' with the desired model name
|
69 |
-
model_name = "Shaleen123/mistrallite_medical_qa"
|
70 |
-
|
71 |
-
config = PeftConfig.from_pretrained(model_name)
|
72 |
-
model = AutoModelForCausalLM.from_pretrained(model_name)
|
73 |
-
model = PeftModel.from_pretrained(model, model_name)
|
74 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
75 |
-
|
76 |
|
77 |
# λν κΈ°λ‘μ μ μ₯νκΈ° μν λ©λͺ¨λ¦¬λ₯Ό μμ±ν©λλ€.
|
78 |
memory = ConversationBufferMemory(
|
|
|
18 |
import tempfile # μμ νμΌμ μμ±νκΈ° μν λΌμ΄λΈλ¬λ¦¬μ
λλ€.
|
19 |
import os
|
20 |
|
21 |
+
with st.spinner("Loading the model"):
|
22 |
+
model_name = "Shaleen123/mistrallite_medical_qa"
|
23 |
+
|
24 |
+
config = PeftConfig.from_pretrained(model_name)
|
25 |
+
model = AutoModelForCausalLM.from_pretrained(model_name)
|
26 |
+
model = PeftModel.from_pretrained(model, model_name)
|
27 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
28 |
+
|
29 |
+
|
30 |
|
31 |
# PDF λ¬Έμλ‘λΆν° ν
μ€νΈλ₯Ό μΆμΆνλ ν¨μμ
λλ€.
|
32 |
def get_pdf_text(pdf_docs):
|
|
|
74 |
return vectorstore # μμ±λ λ²‘ν° μ€ν μ΄λ₯Ό λ°νν©λλ€.
|
75 |
|
76 |
def get_conversation_chain(vectorstore):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
77 |
|
78 |
# λν κΈ°λ‘μ μ μ₯νκΈ° μν λ©λͺ¨λ¦¬λ₯Ό μμ±ν©λλ€.
|
79 |
memory = ConversationBufferMemory(
|