Unizomby commited on
Commit
cd8cf32
·
verified ·
1 Parent(s): d0d6df7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- # Load your OpenAI API key securely
7
  os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
8
 
9
  # Load documents and create index
10
- docs = SimpleDirectoryReader("documents").load_data()
11
- index = VectorStoreIndex.from_documents(docs, service_context=ServiceContext.from_defaults(llm=OpenAI()))
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):