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 = []
|