la04 commited on
Commit
20d1030
·
verified ·
1 Parent(s): 9ff3f09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -3,6 +3,7 @@ import numpy as np
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
  import fitz # PyMuPDF für die Textextraktion aus PDFs
 
6
 
7
  # Schritt 1: Lade das Modell für die Embeddings
8
  model = SentenceTransformer('all-MiniLM-L6-v2')
@@ -34,9 +35,10 @@ def search_documents(query, index, documents):
34
 
35
  # Schritt 4: Gesamtprozess (Fragebeantwortung)
36
  def chatbot_response(pdf, question):
37
- # Speichern der hochgeladenen PDF
38
  pdf_path = "uploaded_pdf.pdf"
39
- pdf.save(pdf_path)
 
40
 
41
  # Textextraktion aus der PDF
42
  text_pages = extract_text_from_pdf(pdf_path)
 
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
  import fitz # PyMuPDF für die Textextraktion aus PDFs
6
+ import os
7
 
8
  # Schritt 1: Lade das Modell für die Embeddings
9
  model = SentenceTransformer('all-MiniLM-L6-v2')
 
35
 
36
  # Schritt 4: Gesamtprozess (Fragebeantwortung)
37
  def chatbot_response(pdf, question):
38
+ # Speichern der hochgeladenen PDF als temporäre Datei
39
  pdf_path = "uploaded_pdf.pdf"
40
+ with open(pdf_path, "wb") as f:
41
+ f.write(pdf.read()) # Schreibe die PDF-Daten in eine temporäre Datei
42
 
43
  # Textextraktion aus der PDF
44
  text_pages = extract_text_from_pdf(pdf_path)