Spaces:
Running
on
Zero
Running
on
Zero
Add sam2 intialization to the start
Browse files
app.py
CHANGED
|
@@ -13,6 +13,7 @@ from PIL import Image
|
|
| 13 |
import tempfile
|
| 14 |
import gc
|
| 15 |
from datetime import datetime
|
|
|
|
| 16 |
|
| 17 |
from addit_flux_pipeline import AdditFluxPipeline
|
| 18 |
from addit_flux_transformer import AdditFluxTransformer2DModel
|
|
@@ -48,6 +49,9 @@ try:
|
|
| 48 |
pipe.scheduler = AdditFlowMatchEulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 49 |
|
| 50 |
print("Model initialized successfully!")
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
except Exception as e:
|
| 53 |
print(f"Error initializing model: {str(e)}")
|
|
|
|
| 13 |
import tempfile
|
| 14 |
import gc
|
| 15 |
from datetime import datetime
|
| 16 |
+
from sam2.sam2_image_predictor import SAM2ImagePredictor
|
| 17 |
|
| 18 |
from addit_flux_pipeline import AdditFluxPipeline
|
| 19 |
from addit_flux_transformer import AdditFluxTransformer2DModel
|
|
|
|
| 49 |
pipe.scheduler = AdditFlowMatchEulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 50 |
|
| 51 |
print("Model initialized successfully!")
|
| 52 |
+
|
| 53 |
+
print("Initialization SAM model:")
|
| 54 |
+
sam = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
|
| 55 |
|
| 56 |
except Exception as e:
|
| 57 |
print(f"Error initializing model: {str(e)}")
|