Translsis commited on
Commit
09d9de2
·
verified ·
1 Parent(s): af4ca2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -87,19 +87,19 @@ print("⏳ Loading SAM3 Models permanently into memory...")
87
  try:
88
  # 1. Load Image Segmentation Model (Text)
89
  print(" ... Loading Image Text Model")
90
- IMG_MODEL = Sam3Model.from_pretrained("facebook/sam3").to(device)
91
- IMG_PROCESSOR = Sam3Processor.from_pretrained("facebook/sam3")
92
 
93
  # 2. Load Image Tracker Model (Click)
94
  print(" ... Loading Image Tracker Model")
95
- TRK_MODEL = Sam3TrackerModel.from_pretrained("facebook/sam3").to(device)
96
- TRK_PROCESSOR = Sam3TrackerProcessor.from_pretrained("facebook/sam3")
97
 
98
  # 3. Load Video Segmentation Model
99
  print(" ... Loading Video Model")
100
  # Using bfloat16 for video to optimize VRAM
101
- VID_MODEL = Sam3VideoModel.from_pretrained("facebook/sam3").to(device, dtype=torch.bfloat16)
102
- VID_PROCESSOR = Sam3VideoProcessor.from_pretrained("facebook/sam3")
103
 
104
  print("✅ All Models loaded successfully!")
105
 
 
87
  try:
88
  # 1. Load Image Segmentation Model (Text)
89
  print(" ... Loading Image Text Model")
90
+ IMG_MODEL = Sam3Model.from_pretrained("DiffusionWave/sam3").to(device)
91
+ IMG_PROCESSOR = Sam3Processor.from_pretrained("DiffusionWave/sam3")
92
 
93
  # 2. Load Image Tracker Model (Click)
94
  print(" ... Loading Image Tracker Model")
95
+ TRK_MODEL = Sam3TrackerModel.from_pretrained("DiffusionWave/sam3").to(device)
96
+ TRK_PROCESSOR = Sam3TrackerProcessor.from_pretrained("DiffusionWave/sam3")
97
 
98
  # 3. Load Video Segmentation Model
99
  print(" ... Loading Video Model")
100
  # Using bfloat16 for video to optimize VRAM
101
+ VID_MODEL = Sam3VideoModel.from_pretrained("DiffusionWave/sam3").to(device, dtype=torch.bfloat16)
102
+ VID_PROCESSOR = Sam3VideoProcessor.from_pretrained("DiffusionWave/sam3")
103
 
104
  print("✅ All Models loaded successfully!")
105