Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# model_name = "dwojcik/gpt2-large-fine-tuned-context-256"
-model_name = "gpt2-large"
+model_name = "dwojcik/gpt2-large-fine-tuned-context-256"
+# model_name = "gpt2-large"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 model.generation_config.temperature = 2.0
 tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="right")