thienphuc12339 commited on
Commit
e7a1f4a
·
verified ·
1 Parent(s): e68f211

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ MODEL_PRESETS = {
22
  "dsta_slr": {
23
  "model": ModelConfig(
24
  arch="dsta_slr",
25
- pretrained="vsltranslation/dsta_slr_joint_motion_v3_0",
26
  ),
27
  "inference": InferenceConfig(
28
  source="upload", # Sử dụng upload, không webcam
@@ -52,7 +52,7 @@ MODEL_PRESETS = {
52
  "spoter": {
53
  "model": ModelConfig(
54
  arch="spoter",
55
- pretrained="vsltranslation/spoter_v3.0",
56
  ),
57
  "inference": InferenceConfig(
58
  source="upload",
 
22
  "dsta_slr": {
23
  "model": ModelConfig(
24
  arch="dsta_slr",
25
+ pretrained="models/dsta_slr_joint_motion_v3_0.onnx",
26
  ),
27
  "inference": InferenceConfig(
28
  source="upload", # Sử dụng upload, không webcam
 
52
  "spoter": {
53
  "model": ModelConfig(
54
  arch="spoter",
55
+ pretrained="models/spoter_v3.0.onnx",
56
  ),
57
  "inference": InferenceConfig(
58
  source="upload",