Update app.py

app.py

@@ -9,7 +9,8 @@ class AppModel:
     def __init__(self):
         self.tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
         self.tokenizer.pad_token = self.tokenizer.eos_token
-        self.model = AutoModelForCausalLM.from_pretrained("./output/model/checkpoint-1000/")
+        self.model = AutoModelForCausalLM.from_pretrained("gpt2-medium")
+
 
     def generate_plot(self, prompt: str):
         inputs = self.tokenizer(prompt, return_tensors="pt").to("cpu")