Update app.py

app.py

@@ -485,7 +485,8 @@ def main(input_text, max_new_tokens):
     else:
         device = torch.device("cpu")
 
-    weights = torch.load("model_and_optimizer.pth", map_location=torch.device(device))
+    # weights = torch.load("model_and_optimizer.pth", map_location=torch.device(device))
+    weights = torch.load("model_and_optimizer.pth", weights_only=False)
 
     model = GPTModel({
     "vocab_size": 50257,   # Vocabulary size