Add application file

app.py

@@ -20,9 +20,9 @@ login(token='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', add_to_git_credential=True)
 st.title("Code Generation")
 huggingface_dataset_name = "red1xe/code_instructions"
 dataset = load_dataset(huggingface_dataset_name)
-model_name='gpt2'
-tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoder")
-original_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder")
+model_name='bigcode/starcoder'
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+original_model = AutoModelForCausalLM.from_pretrained(model_name)
 
 x = st.slider(label='Select a sample', min_value=0, max_value=1000, value=500, step=10)
 if st.button("Show Sample"):