Update app.py

app.py

@@ -16,7 +16,7 @@ config = GPT2Config.from_pretrained("distilgpt2")
 model = GPT2LMHeadModel(config)
 
 # Load the weights from the pytorch_model.bin file
-model_path = "./pytorch_model.bin"  # Path to your local model file
+model_path = "./pytorch_model_20.bin"  # Path to local model file
 state_dict = torch.load(model_path, map_location=device)  # Load the state_dict
 model.load_state_dict(state_dict)  # Load the state dict into the model