Upload folder using huggingface_hub

app.py

@@ -13,6 +13,7 @@ import gradio as gr
 import matplotlib as mpl
 from huggingface_hub import hf_hub_download
 import onnxruntime as ort
+import spaces
 
 model_dict = {
     "BallNet": "asRobotics/ballnet",
@@ -77,6 +78,7 @@ def make_double_sided(mesh: trimesh.Trimesh) -> trimesh.Trimesh:
     return mesh_double
 
 
+@spaces.GPU
 def predict_from_motion(dx, dy, dz, rx, ry, rz, model_name):
     motion = np.array(
         [dx, dy, dz, rx / 180.0 * np.pi, ry / 180.0 * np.pi, rz / 180.0 * np.pi],
@@ -92,7 +94,8 @@ def predict_from_motion(dx, dy, dz, rx, ry, rz, model_name):
         force.reshape(1, -1),
         columns=["Fx (N)", "Fy (N)", "Fz (N)", "Tx (Nmm)", "Ty (Nmm)", "Tz (Nmm)"],
     )
-    df = df.style.format("{:.3f}")
+    # Round to 3 decimal places for display (avoids pickle error with style objects)
+    df = df.round(3)
 
     delta_vertices = np.zeros_like(vertices_dict[model_name])
     delta_vertices[def_index_dict[model_name] - 1] += nodes.reshape(-1, 3)
@@ -120,6 +123,7 @@ def predict_from_motion(dx, dy, dz, rx, ry, rz, model_name):
 
     tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".glb")
     mesh.export(tmp_file.name, file_type="glb")
+    tmp_file.close()
 
     return df, tmp_file.name
 

requirements.txt

@@ -6,3 +6,4 @@ pandas
 matplotlib
 trimesh
 scipy
+spaces