ofermend commited on
Commit
ca0390a
·
verified ·
1 Parent(s): 60ef7a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from omegaconf import OmegaConf
2
  from query import VectaraQuery
3
  import os
 
4
 
5
  import streamlit as st
6
  from PIL import Image
@@ -39,7 +40,7 @@ def launch_bot():
39
  if 'cfg' not in st.session_state:
40
  cfg = OmegaConf.create({
41
  'customer_id': str(os.environ['customer_id']),
42
- 'corpus_ids': [cid.strip() for cid in os.environ['corpus_ids'].split(',') if cid.strip()],
43
  'api_key': str(os.environ['api_key']),
44
  'title': os.environ['title'],
45
  'description': os.environ['description'],
 
1
  from omegaconf import OmegaConf
2
  from query import VectaraQuery
3
  import os
4
+ import json
5
 
6
  import streamlit as st
7
  from PIL import Image
 
40
  if 'cfg' not in st.session_state:
41
  cfg = OmegaConf.create({
42
  'customer_id': str(os.environ['customer_id']),
43
+ 'corpus_ids': [str(cid) for cid in json.loads(os.environ['corpus_ids'])],
44
  'api_key': str(os.environ['api_key']),
45
  'title': os.environ['title'],
46
  'description': os.environ['description'],