gsacco15 commited on
Commit
5e61a90
·
verified ·
1 Parent(s): feeacb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,13 +3,12 @@ import torch
3
  import numpy as np
4
  from PIL import Image
5
  from transformers import Sam3Processor, Sam3Model
6
- import json
7
 
8
  # Load model on startup
9
  print("Loading SAM 3 model...")
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
- model = Sam3Model.from_pretrained("facebook/sam3").to(device)
12
- processor = Sam3Processor.from_pretrained("facebook/sam3")
13
  print(f"Model loaded on {device}")
14
 
15
  # Reference dimensions (inches)
 
3
  import numpy as np
4
  from PIL import Image
5
  from transformers import Sam3Processor, Sam3Model
 
6
 
7
  # Load model on startup
8
  print("Loading SAM 3 model...")
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
+ model = Sam3Model.from_pretrained("facebook/sam3", token=True).to(device)
11
+ processor = Sam3Processor.from_pretrained("facebook/sam3", token=True)
12
  print(f"Model loaded on {device}")
13
 
14
  # Reference dimensions (inches)