Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,14 @@
|
|
1 |
import gradio as gr
|
2 |
-
from llama_index import SimpleDirectoryReader, VectorStoreIndex, ServiceContext
|
3 |
-
from llama_index.llms import OpenAI
|
4 |
import os
|
|
|
|
|
5 |
|
6 |
-
#
|
7 |
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
8 |
|
9 |
# Load documents and create index
|
10 |
-
|
11 |
-
index = VectorStoreIndex.from_documents(
|
12 |
query_engine = index.as_query_engine()
|
13 |
|
14 |
def chatbot_response(message):
|
|
|
1 |
import gradio as gr
|
|
|
|
|
2 |
import os
|
3 |
+
from llama_index.core import SimpleDirectoryReader, VectorStoreIndex, ServiceContext
|
4 |
+
from llama_index.llms.openai import OpenAI
|
5 |
|
6 |
+
# Set your OpenAI API key
|
7 |
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
8 |
|
9 |
# Load documents and create index
|
10 |
+
documents = SimpleDirectoryReader("documents").load_data()
|
11 |
+
index = VectorStoreIndex.from_documents(documents, service_context=ServiceContext.from_defaults(llm=OpenAI()))
|
12 |
query_engine = index.as_query_engine()
|
13 |
|
14 |
def chatbot_response(message):
|