Update app

app.py

@@ -26,11 +26,11 @@ model_dictionary = {
 
 st.title("Text Simplification Model")
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource()
 def load(dataset_name, model_variant_name):
     return pipeline(
         "text2text-generation", 
-        model="lucadiliello/bart-small" # model_dictionary[dataset_name][model_variant_name]
+        model=model_dictionary[dataset_name][model_variant_name]
         )
 
 def predict(text, pipeline):
@@ -77,7 +77,7 @@ st.text_area("Text to Simplify:", key="text", height=275)
 
 # Load model and run inference
 if st.button("Simplify!"):
-    # Number 1
+    # # Number 1
     # # tokenizer_baseline, model_baseline = load(dataset_option, "baseline")
     # # model_outputs_baseline = predict(st.session_state.text, model_baseline, tokenizer_baseline)[0]
     
@@ -95,7 +95,7 @@ if st.button("Simplify!"):
     #     )
     # st.write(f"Baseline: {clean(model_outputs_baseline)}")
 
-    # Number 2
+    # # Number 2
     # tokenizer_ul, model_ul = load(dataset_option, "ul")
     # model_outputs_ul = predict(st.session_state.text, model_ul, tokenizer_ul)[0]