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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,14 +1,19 @@
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):
 
1
  import gradio as gr
2
  import os
3
+ from llama_index.core import SimpleDirectoryReader, VectorStoreIndex, Settings
4
  from llama_index.llms.openai import OpenAI
5
 
6
+ # Set OpenAI API key from environment
7
  os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
8
 
9
+ # Configure the default LLM
10
+ Settings.llm = OpenAI()
11
+
12
+ # Load documents and build index
13
  documents = SimpleDirectoryReader("documents").load_data()
14
+ index = VectorStoreIndex.from_documents(documents)
15
+
16
+ # Create query engine
17
  query_engine = index.as_query_engine()
18
 
19
  def chatbot_response(message):