Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -75,13 +75,12 @@ steel_blue_theme = SteelBlueTheme()
|
|
| 75 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 76 |
print(f"Using device: {device}")
|
| 77 |
|
| 78 |
-
# --- Model Loading ---
|
| 79 |
-
# Using the facebook/sam3 model as requested
|
| 80 |
try:
|
| 81 |
print("Loading SAM3 Model and Processor...")
|
| 82 |
model = Sam3Model.from_pretrained("facebook/sam3").to(device)
|
| 83 |
processor = Sam3Processor.from_pretrained("facebook/sam3")
|
| 84 |
print("Model loaded successfully.")
|
|
|
|
| 85 |
except Exception as e:
|
| 86 |
print(f"Error loading model: {e}")
|
| 87 |
print("Ensure you have the correct libraries installed and access to the model.")
|
|
|
|
| 75 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 76 |
print(f"Using device: {device}")
|
| 77 |
|
|
|
|
|
|
|
| 78 |
try:
|
| 79 |
print("Loading SAM3 Model and Processor...")
|
| 80 |
model = Sam3Model.from_pretrained("facebook/sam3").to(device)
|
| 81 |
processor = Sam3Processor.from_pretrained("facebook/sam3")
|
| 82 |
print("Model loaded successfully.")
|
| 83 |
+
|
| 84 |
except Exception as e:
|
| 85 |
print(f"Error loading model: {e}")
|
| 86 |
print("Ensure you have the correct libraries installed and access to the model.")
|