DHEIVER commited on
Commit
31fe207
·
verified ·
1 Parent(s): 9740646

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
- from langchain.document_loaders import PyPDFLoader
3
  from langchain.text_splitter import RecursiveCharacterTextSplitter
4
- from langchain.embeddings import HuggingFaceEmbeddings
5
- from langchain.vectorstores import FAISS
6
  from langchain.chains import RetrievalQA
7
- from langchain.llms import HuggingFaceHub
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.name)
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="file"),
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"),