Update app.py

app.py

@@ -9,9 +9,8 @@ from model_gpt import GPT, GPTConfig
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 #load the model
-out_dir = '/content/drive/MyDrive/ERA_V2/Assignment19/nanoGPT/nanoGPT/out-shakespeare-char/'
-ckpt_path = os.path.join(out_dir, 'ckpt.pt')
-checkpoint = torch.load(ckpt_path, map_location=device)
+
+checkpoint = torch.load('ckpt.pt', map_location=device)
 gptconf = GPTConfig(**checkpoint['model_args'])
 model = GPT(gptconf)
 state_dict = checkpoint['model']
@@ -22,7 +21,7 @@ for k,v in list(state_dict.items()):
 model.load_state_dict(state_dict)
 
 #load the dataset
-with open('/content/drive/MyDrive/ERA_V2/Assignment19/input.txt', 'r', encoding='utf-8') as f:
+with open('input.txt', 'r', encoding='utf-8') as f:
     text = f.read()
 
 # here are all the unique characters that occur in this text