Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import CharacterTextSplitter
-from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 from langchain.chains import RetrievalQA
-from langchain.llms.base import LLM  # Basis-Klasse, um einen Wrapper zu erstellen
+from langchain.llms.base import LLM
 from transformers import pipeline
 
-# LeoLM-Wrapper-Klasse, die das LeoLM-Modell via Transformers-Pipeline nutzt
+# LeoLM-Wrapper-Klasse, die das LeoLM-Modell über die Transformers-Pipeline nutzt
 class LeoLM(LLM):
     def __init__(self, max_new_tokens=512, temperature=0.5):
         self.pipeline = pipeline("text-generation", model="LeoLM/leo-mistral-hessianai-7b")
@@ -21,8 +21,12 @@ class LeoLM(LLM):
     @property
     def _identifying_params(self):
         return {"model": "LeoLM/leo-mistral-hessianai-7b"}
+    
+    @property
+    def _llm_type(self):
+        return "custom_leolm"
 
-# PDF wird beim Start automatisch geladen
+# PDF wird beim Start automatisch geladen und verarbeitet
 loader = PyPDFLoader("TrendingMedia_ChatbotBasis_FINAL.pdf")
 documents = loader.load()
 splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=100)