Update app.py

app.py

@@ -13,8 +13,6 @@ Original file is located at
 
 
 import torch
-torch.cuda.is_available()
-
 import glob
 import math
 import sys
@@ -114,6 +112,7 @@ def tsaigpt(start:str , max_new_tokens = 300, num_samples =2, tokeniser= token):
     model = GPT(config)
 
     load_checkpoint(fabric, model, checkpoint_path)
+    print(model)
     model.eval()
     model.to(device)
     if compile: