Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
-
from
|
3 |
-
from
|
4 |
-
from
|
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
|
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:
|