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
|