fastx commited on
Commit
d822b20
·
verified ·
1 Parent(s): dd78d61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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="emilyalsentzer/Bio_ClinicalBERT",
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 = []