Bc-AI Keeby-smilyai commited on
Commit
728df68
·
verified ·
1 Parent(s): 65f0a83

Update app.py (#2)

Browse files

- Update app.py (71f8cdb6f5cb1a54334c0d34cc9d1a5c584ac700)


Co-authored-by: Boning c <Keeby-smilyai@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -317,9 +317,9 @@ print("✅ Tokenizer ready")
317
  # Model Registry
318
  MODEL_REGISTRY = [
319
  ("SAM-X-1-Large", "Smilyai-labs/Sam-1x-instruct", "ckpt.weights.h5", None),
320
- ("SAM-X-1-Fast ⚡ (BETA)", "Smilyai-labs/Sam-X-1-fast", "sam1_fast.weights.h5", "sam1_fast_config.json"),
321
- ("SAM-X-1-Mini 🚀 (BETA)", "Smilyai-labs/Sam-X-1-Mini", "sam1_mini.weights.h5", "sam1_mini_config.json"),
322
- ("SAM-X-1-Nano ⚡⚡ (BETA)", "Smilyai-labs/Sam-X-1-Nano", "sam1_nano.weights.h5", "sam1_nano_config.json"),
323
  ]
324
 
325
  available_models = {}
 
317
  # Model Registry
318
  MODEL_REGISTRY = [
319
  ("SAM-X-1-Large", "Smilyai-labs/Sam-1x-instruct", "ckpt.weights.h5", None),
320
+ ("SAM-X-1-Fast ⚡ (BETA)", "Smilyai-labs/Sam-X-1-fast", "sam1_fast_finetuned.weights.h5", "sam1_fast_finetuned_config.json"),
321
+ ("SAM-X-1-Mini 🚀 (BETA)", "Smilyai-labs/Sam-X-1-Mini", "sam1_mini.weights_finetuned.h5", "sam1_mini_finetuned_config.json"),
322
+ ("SAM-X-1-Nano ⚡⚡ (BETA)", "Smilyai-labs/Sam-X-1-Nano", "sam1_nano_finetuned.weights.h5", "sam1_nano_finetuned_config.json"),
323
  ]
324
 
325
  available_models = {}