Update app.py

app.py

@@ -441,7 +441,8 @@ def process_hdf5_file(uploaded_file, percentage):
         # Step 4: Load the model from lwm_model module
         device = 'cuda' if torch.cuda.is_available() else 'cpu'
         print(f"Loading the LWM model on {device}...")
-        model = lwm_model.LWM.from_pretrained(device=device).to(torch.float32)
+        model = lwm_model.LWM.from_pretrained(device=device)
+        model = model.float()
 
         # Step 5: Load the HDF5 file and extract the channels and labels
         with h5py.File(uploaded_file.name, 'r') as f: