Update app.py

app.py

@@ -42,9 +42,9 @@ def init_model():
     model = load_model(model, args.load_model_path)
     model.eval()
 
-    for name, parameter in model.named_parameters():
-        print(name)
-        print(parameter)
+    # for name, parameter in model.named_parameters():
+    #     print(name)
+    #     print(parameter)
     print(torch.cuda.max_memory_allocated() / 1024 ** 3)
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model.to(device)