Update app.py

app.py

@@ -97,12 +97,13 @@ def run_inference(uploaded_file):
     sys.stdout = capture  # Redirect print statements to capture
 
     try:
-        # Load the HDF5 file and extract channels and labels
+        # Step 5: Load the HDF5 file and extract the channels and labels
         with h5py.File(uploaded_file.name, 'r') as f:
-            channels = np.array(f['channels'])  # Assuming 'channels' dataset in the HDF5 file
-            labels = np.array(f['labels'])  # Assuming 'labels' dataset in the HDF5 file
+            channels = np.asarray(f['channels'])  # Use np.asarray to avoid DeprecationWarning
+            labels = np.asarray(f['labels'])      # Use np.asarray to avoid DeprecationWarning
         print(f"Loaded dataset with {channels.shape[0]} samples.")
 
+
         # Run the tokenization and model inference
         model_repo_url = "https://huggingface.co/sadjadalikhani/LWM"
         model_repo_dir = "./LWM"