Copain22 commited on
Commit
0f9d9fe
·
verified ·
1 Parent(s): 53c41f2

update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import torch, os
2
  import gradio as gr
 
3
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
4
  from llama_index.core.memory import ChatMemoryBuffer
5
  from llama_index.llms.huggingface import HuggingFaceLLM
@@ -7,10 +8,10 @@ from llama_index.embeddings.langchain import LangchainEmbedding
7
  from langchain.embeddings.huggingface import HuggingFaceEmbeddings
8
  from llama_index.core import PromptTemplate
9
 
10
- # ---------- 1. Load menu PDF ----------
11
- docs = SimpleDirectoryReader(input_files=["cafeoneeleven.pdf"]).load_data()
12
 
13
- # ---------- 2. LLM + embeddings ----------
 
 
14
  SYSTEM_PROMPT = """
15
  You are a friendly café assistant for Café Eleven. Your job is to:
16
  1. Greet the customer warmly
 
1
  import torch, os
2
  import gradio as gr
3
+ from pathlib import Path
4
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
5
  from llama_index.core.memory import ChatMemoryBuffer
6
  from llama_index.llms.huggingface import HuggingFaceLLM
 
8
  from langchain.embeddings.huggingface import HuggingFaceEmbeddings
9
  from llama_index.core import PromptTemplate
10
 
 
 
11
 
12
+ pdf_files = [str(p) for p in Path(".").glob("*.pdf")]
13
+ docs = SimpleDirectoryReader(input_files=pdf_files).load_data()
14
+
15
  SYSTEM_PROMPT = """
16
  You are a friendly café assistant for Café Eleven. Your job is to:
17
  1. Greet the customer warmly