Update app.py
Browse files
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("
|
| 91 |
-
IMG_PROCESSOR = Sam3Processor.from_pretrained("
|
| 92 |
|
| 93 |
# 2. Load Image Tracker Model (Click)
|
| 94 |
print(" ... Loading Image Tracker Model")
|
| 95 |
-
TRK_MODEL = Sam3TrackerModel.from_pretrained("
|
| 96 |
-
TRK_PROCESSOR = Sam3TrackerProcessor.from_pretrained("
|
| 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("
|
| 102 |
-
VID_PROCESSOR = Sam3VideoProcessor.from_pretrained("
|
| 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 |
|