added torch lib
Browse files- app.py +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
|
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
|
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
|