Add application file

app.py

@@ -10,8 +10,8 @@ import streamlit as st
 st.title('Code Generation')
 huggingface_dataset_name = "red1xe/code_instructions"
 dataset = load_dataset(huggingface_dataset_name)
-model_name='google/flan-t5-base'
-original_model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
+model_name='gpt2'
+original_model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 x = st.slider('Select a sample', 0, 1000, 200)