Update app.py

app.py

@@ -1,14 +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")
-    model = AutoModelWithLMHead.from_pretrained("GPT-python")
+    tokenizer = AutoTokenizer.from_pretrained("./")
+    model = AutoModelWithLMHead.from_pretrained("./")
 
     input_ids = tokenizer.encode(inp, return_tensors="pt")
     beam_output = model.generate(input_ids,