Update app.py

app.py

@@ -6,8 +6,7 @@ import torch
 # Load tokenizer
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 tokenizer.pad_token = tokenizer.eos_token
-model.load_state_dict(torch.load("Conv_GPT.pth", map_location=torch.device('cpu')))
-model.eval()
+
 
 # Load model
 model = TransformerModel(
@@ -17,6 +16,8 @@ model = TransformerModel(
     num_heads=16,
     dropout=0.1
 )
+model.load_state_dict(torch.load("Conv_GPT.pth", map_location=torch.device('cpu')))
+model.eval()
 
 
 # Define generation function