Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,8 @@ import torch
|
|
6 |
import streamlit as st
|
7 |
from dotenv import load_dotenv
|
8 |
from langchain_groq import ChatGroq
|
9 |
-
from langchain_community.embeddings import HuggingFaceEmbeddings
|
|
|
10 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
11 |
from langchain_community.vectorstores import FAISS
|
12 |
from langchain_core.documents import Document
|
@@ -54,11 +55,12 @@ except Exception as e:
|
|
54 |
|
55 |
# Set up embeddings for clinical context (Bio_ClinicalBERT)
|
56 |
embeddings = HuggingFaceEmbeddings(
|
57 |
-
model_name="
|
58 |
model_kwargs={"device": device}
|
59 |
)
|
60 |
|
61 |
|
|
|
62 |
def load_clinical_data():
|
63 |
"""Load both flowcharts and patient cases"""
|
64 |
docs = []
|
|
|
6 |
import streamlit as st
|
7 |
from dotenv import load_dotenv
|
8 |
from langchain_groq import ChatGroq
|
9 |
+
#from langchain_community.embeddings import HuggingFaceEmbeddings
|
10 |
+
from langchain_huggingface import HuggingFaceEmbeddings
|
11 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
12 |
from langchain_community.vectorstores import FAISS
|
13 |
from langchain_core.documents import Document
|
|
|
55 |
|
56 |
# Set up embeddings for clinical context (Bio_ClinicalBERT)
|
57 |
embeddings = HuggingFaceEmbeddings(
|
58 |
+
model_name="pritamdeka/BioBERT-mnli-snli-scinli-scitail-mednli-stsb",
|
59 |
model_kwargs={"device": device}
|
60 |
)
|
61 |
|
62 |
|
63 |
+
|
64 |
def load_clinical_data():
|
65 |
"""Load both flowcharts and patient cases"""
|
66 |
docs = []
|