zlmqi commited on
Commit
a80eae4
1 Parent(s): 2eb80d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -3,13 +3,16 @@ import json
3
  import gradio as gr
4
  import os
5
 
 
 
 
6
  # retrieve secret key from HF settings
7
  #openai.api_key = os.getenv("OPENAI_API_KEY")
8
 
9
  # rebuild storage context and load knowledge index
10
  from llama_index import StorageContext, load_index_from_storage
11
- storage_context = StorageContext.from_defaults('./index/')
12
- index = load_index_from_storage(storage_context, service_context=service_context)
13
 
14
  class Chatbot:
15
  def __init__(self, api_key, index):
 
3
  import gradio as gr
4
  import os
5
 
6
+ from llama_index import set_global_service_context
7
+ set_global_service_context(service_context)
8
+
9
  # retrieve secret key from HF settings
10
  #openai.api_key = os.getenv("OPENAI_API_KEY")
11
 
12
  # rebuild storage context and load knowledge index
13
  from llama_index import StorageContext, load_index_from_storage
14
+ storage_context = StorageContext.from_defaults(persist_dir='./index/')
15
+ index = load_index_from_storage(storage_context)
16
 
17
  class Chatbot:
18
  def __init__(self, api_key, index):