Add application file

app.py

@@ -10,12 +10,9 @@ import streamlit as st
 st.title('Code Generation')
 huggingface_dataset_name = "red1xe/code_instructions"
 dataset = load_dataset(huggingface_dataset_name)
-
-if st.button("Load Model"):
-    with st.spinner('Loading Model...'):
-        model_name='google/flan-t5-base'
-        original_model = AutoModelForSeq2SeqLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
-        tokenizer = AutoTokenizer.from_pretrained(model_name)
+model_name='google/flan-t5-base'
+original_model = AutoModelForSeq2SeqLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 x = st.slider('Select a sample', 0, 1000, 200)
 if st.button("Show Sample"):