dipta007 commited on
Commit
c34db09
·
1 Parent(s): 35b9ece

added torch lib

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -32,7 +32,7 @@ if "messages" not in st.session_state:
32
  if len(st.session_state.messages) == 0 and "id" in st.query_params:
33
  with st.spinner("Loading chat..."):
34
  id = st.query_params["id"]
35
- data = get_file(id, 'llm-007')
36
  obj = pickle.loads(data)
37
  for k, v in obj.items():
38
  st.session_state[k] = v
@@ -44,7 +44,7 @@ def share():
44
  if k in st.session_state:
45
  obj[k] = st.session_state[k]
46
  data = pickle.dumps(obj)
47
- id = upload_file(data, 'llm-007')
48
  url = f"https://umbc-nlp-llm.hf.space/?id={id}"
49
  st.markdown(f"[share](/?id={id})")
50
  st.success(f"Share URL: {url}")
 
32
  if len(st.session_state.messages) == 0 and "id" in st.query_params:
33
  with st.spinner("Loading chat..."):
34
  id = st.query_params["id"]
35
+ data = get_file(id)
36
  obj = pickle.loads(data)
37
  for k, v in obj.items():
38
  st.session_state[k] = v
 
44
  if k in st.session_state:
45
  obj[k] = st.session_state[k]
46
  data = pickle.dumps(obj)
47
+ id = upload_file(data)
48
  url = f"https://umbc-nlp-llm.hf.space/?id={id}"
49
  st.markdown(f"[share](/?id={id})")
50
  st.success(f"Share URL: {url}")
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  streamlit
2
  boto3
3
- transformers
 
 
1
  streamlit
2
  boto3
3
+ transformers
4
+ torch