Spaces:
Running
on
Zero
Running
on
Zero
update app.py
Browse files
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 |
-
|
|
|
|
|
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
|