shnl commited on
Commit
cd57d9b
·
1 Parent(s): 539a3a1
Files changed (1) hide show
  1. src/streamlit_app.py +3 -0
src/streamlit_app.py CHANGED
@@ -2,6 +2,9 @@ import streamlit as st
2
  import torch
3
  import time
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 
 
5
 
6
  st.set_page_config(page_title="Question Generation", layout="wide")
7
 
 
2
  import torch
3
  import time
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
+ import os
6
+ os.environ["HF_HOME"] = "/tmp/huggingface"
7
+
8
 
9
  st.set_page_config(page_title="Question Generation", layout="wide")
10