GPT-2-with_gpu

Sleeping

sagar007 commited on Jun 28, 2024

Commit

1ffcf64

verified ·

1 Parent(s): e9779bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 model = GPT2LMHeadModel.from_pretrained('gpt2')
 # Load the best model weights
-model.load_state_dict(torch.load('best_model.pth', map_location=torch.device('cpu')))
 # Set the model to evaluation mode
 model.eval()

 model = GPT2LMHeadModel.from_pretrained('gpt2')
 # Load the best model weights
+model.load_state_dict(torch.load('GPT_model.pth', map_location=torch.device('cpu')))
 # Set the model to evaluation mode
 model.eval()