build error debug

app.py

@@ -1,20 +1,15 @@
+import os
+import pickle
 import numpy as np
-import matplotlib.pyplot as plt
-from functools import partial
+
 import torch
 from torch_geometric.nn import knn
+
 import gradio as gr
-import os
-import glob
-import pickle
-import time
-import cv2
-import numpy as np
 import matplotlib.pyplot as plt
-import plotly.graph_objects as go
 import open3d as o3d
 import trimesh
-import gradio
+
 
 
 
@@ -173,7 +168,7 @@ def compute_centroids(data, labels):
     for label in unique_labels:
         mask = labels == label
         points = data[mask]
-        centroid = jnp.mean(points, axis=0)
+        centroid = np.mean(points, axis=0)
         centroids.append(centroid)
     return np.stack(centroids)
 

requirements.txt

@@ -1,5 +1,7 @@
 trimesh
 open3d
+
+torch
 torch_geometric
 torch_scatter
 -f https://data.pyg.org/whl/torch-2.3.0+cpu.html