Update app.py

app.py

@@ -1,6 +1,10 @@
 from transformers import AutoTokenizer, AutoModelWithLMHead
+from urllib.request import urlretrieve
 import gradio as gr
 
+# Loads latest model state from Github
+urlretrieve("https://github.com/equ1/generative_python_transformer/tree/main/GPT-python")
+
 # inference function
 def inference(inp):
     tokenizer = AutoTokenizer.from_pretrained("GPT-python")