Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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):
|