fixing model with eval on deployment

README.md

@@ -1,3 +1,13 @@
+---
+title: Idiomify demo
+emoji: ✍️
+colorFrom: yellow
+colorTo: orange
+sdk: streamlit
+app_file: main_deploy.py
+pinned: false
+---
+
 # Idiomify
 [![Open in Streamlit](https://static.streamlit.io/badges/streamlit_badge_black_white.svg)](https://share.streamlit.io/eubinecto/idiomify/issue_2/main_deploy.py)
 

main_deploy.py

@@ -21,6 +21,7 @@ def fetch_resources() -> Tuple[dict, Idiomifier, BartTokenizer, List[str]]:
 def main():
     # fetch a pre-trained model
     config, model, tokenizer, idioms = fetch_resources()
+    model.eval()
     pipeline = Pipeline(model, tokenizer)
     st.title("Idiomify Demo")
     st.markdown(f"Author: `Eu-Bin KIM`")