Update app.py

app.py

@@ -1,10 +1,10 @@
 import torch
-from model import  GPT, decode
+from model import  BigramLanguageModel, decode
 import gradio as gr
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-model = GPT()
+model = BigramLanguageModel()
 model.load_state_dict(torch.load("./neo_gpt.pth", map_location=device))
 def generate_text(max_new_tokens):
     context = torch.zeros((1, 1), dtype=torch.long)