Update app.py

app.py

@@ -450,8 +450,8 @@ def process_hdf5_file(uploaded_file, percentage):
         print(f"Loading the LWM model on {device}...")
         model = lwm_model.LWM.from_pretrained(device=device)
         model = model.float()
-        for name, param in model.state_dict().items():
-            print(f"Layer: {name} | Weights: {param}")
+        #for name, param in model.state_dict().items():
+        #    print(f"Layer: {name} | Weights: {param}")
 
         # Step 5: Load the HDF5 file and extract the channels and labels
         with h5py.File(uploaded_file.name, 'r') as f: