Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
import gradio as gr
|
2 |
-
from
|
3 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
4 |
-
from
|
5 |
-
from
|
6 |
from langchain.chains import RetrievalQA
|
7 |
-
from
|
8 |
|
9 |
# Configurações
|
10 |
EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
|
@@ -32,7 +32,7 @@ def generate_response(pdf_file, query):
|
|
32 |
return "Erro: Nenhum arquivo PDF foi carregado."
|
33 |
|
34 |
# Carrega e processa o PDF
|
35 |
-
db = load_and_process_pdf(pdf_file
|
36 |
|
37 |
# Configura o modelo de linguagem
|
38 |
llm = HuggingFaceHub(repo_id=LLM_REPO_ID, model_kwargs={"temperature": 0.7, "max_length": 512})
|
@@ -53,7 +53,7 @@ def generate_response(pdf_file, query):
|
|
53 |
iface = gr.Interface(
|
54 |
fn=generate_response,
|
55 |
inputs=[
|
56 |
-
gr.File(label="Upload PDF", type="
|
57 |
gr.Textbox(label="Sua Pergunta")
|
58 |
],
|
59 |
outputs=gr.Textbox(label="Resposta Gerada"),
|
|
|
1 |
import gradio as gr
|
2 |
+
from langchain_community.document_loaders import PyPDFLoader
|
3 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
4 |
+
from langchain_community.embeddings import HuggingFaceEmbeddings
|
5 |
+
from langchain_community.vectorstores import FAISS
|
6 |
from langchain.chains import RetrievalQA
|
7 |
+
from langchain_community.llms import HuggingFaceHub
|
8 |
|
9 |
# Configurações
|
10 |
EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
|
|
|
32 |
return "Erro: Nenhum arquivo PDF foi carregado."
|
33 |
|
34 |
# Carrega e processa o PDF
|
35 |
+
db = load_and_process_pdf(pdf_file)
|
36 |
|
37 |
# Configura o modelo de linguagem
|
38 |
llm = HuggingFaceHub(repo_id=LLM_REPO_ID, model_kwargs={"temperature": 0.7, "max_length": 512})
|
|
|
53 |
iface = gr.Interface(
|
54 |
fn=generate_response,
|
55 |
inputs=[
|
56 |
+
gr.File(label="Upload PDF", type="filepath"), # Corrigido para "filepath"
|
57 |
gr.Textbox(label="Sua Pergunta")
|
58 |
],
|
59 |
outputs=gr.Textbox(label="Resposta Gerada"),
|