Spaces:

pappymu
/

question-gen

Runtime error

pappymu commited on Aug 27, 2022

Commit

4e3bb23

•

1 Parent(s): 0a5c2e0

Update streamlit_app.py

Files changed (1) hide show

streamlit_app.py CHANGED Viewed

@@ -242,7 +242,7 @@ with st.form('my_form'):
 with st.spinner('Loading Model...'):
     model = T5Model
-    best_model_dir = 'checkpoints/t5-chkpt-v2.ckpt'
     best_model = model.load_from_checkpoint(best_model_dir)
     # best_model = model.load_from_checkpoint(callback.best_model_path)
     best_model.freeze()

 with st.spinner('Loading Model...'):
     model = T5Model
+    best_model_dir = 't5-chkpt-v2.ckpt'
     best_model = model.load_from_checkpoint(best_model_dir)
     # best_model = model.load_from_checkpoint(callback.best_model_path)
     best_model.freeze()