zlmqi commited on
Commit
5d0c781
1 Parent(s): c73977c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -26,8 +26,8 @@ from llama_index.llms.openai import OpenAI
26
  from datetime import datetime;
27
  import socket;
28
 
29
- # access data stored in datasets
30
- from datasets import load_dataset
31
 
32
  # deprecated
33
  storage_context = StorageContext.from_defaults(persist_dir='./')
@@ -42,8 +42,9 @@ class Chatbot:
42
  self.index = index
43
  openai.api_key = api_key
44
  self.chat_history = []
45
- #self.history_file = f"zlmqi/index/chat_log.json"
46
- self.history_file = load_dataset("json", data_files="zlmqi/index/chat_log.json")
 
47
 
48
  def generate_response(self, user_input):
49
  query_engine = index.as_query_engine(llm=llm_predictor)
 
26
  from datetime import datetime;
27
  import socket;
28
 
29
+ # access /data folder of persistent storage
30
+ from pathlib import Path
31
 
32
  # deprecated
33
  storage_context = StorageContext.from_defaults(persist_dir='./')
 
42
  self.index = index
43
  openai.api_key = api_key
44
  self.chat_history = []
45
+
46
+ # access chat log file in /data folder (persistent storage)
47
+ self.history_file = f"/data/chat_log.json"
48
 
49
  def generate_response(self, user_input):
50
  query_engine = index.as_query_engine(llm=llm_predictor)