Kathirsci commited on
Commit
ffaaa3a
·
verified ·
1 Parent(s): 43c5fe0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
- from langchain.document_loaders import PyPDFLoader
3
- from langchain.vectorstores import FAISS
4
- from langchain.llms import HuggingFaceLLM
5
  from langchain.chains import MapReduceChain
6
  from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from langchain.embeddings import HuggingFaceEmbeddings
@@ -10,7 +10,7 @@ import os
10
 
11
  # Load model and embeddings
12
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", device="cpu")
13
- llm = HuggingFaceLLM.from_pretrained("your-model")
14
 
15
  def process_pdf_and_summarize(file):
16
  try:
 
1
  import gradio as gr
2
+ from langchain_community.document_loaders import PyPDFLoader
3
+ from langchain_community.vectorstores import FAISS
4
+ from langchain_community.llms import HuggingFaceLLM
5
  from langchain.chains import MapReduceChain
6
  from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from langchain.embeddings import HuggingFaceEmbeddings
 
10
 
11
  # Load model and embeddings
12
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", device="cpu")
13
+ llm = HuggingFaceLLM(model_name="llama2")
14
 
15
  def process_pdf_and_summarize(file):
16
  try: