Spaces:
Build error
Build error
123LETSPLAY
commited on
Commit
•
0e9a7e8
1
Parent(s):
20f29bf
Create requirements.txt
Browse files- requirements.txt +3 -9
requirements.txt
CHANGED
@@ -1,9 +1,3 @@
|
|
1 |
-
#
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
predictor = SAM2ImagePredictor.from_pretrained(facebook/sam2.1-hiera-large)
|
6 |
-
|
7 |
-
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
|
8 |
-
predictor.set_image(<your_image>)
|
9 |
-
masks, _, _ = predictor.predict(<input_prompts>)
|
|
|
1 |
+
torch==2.0.1 # PyTorch for model execution
|
2 |
+
gradio==3.24.1 # Gradio for web interface
|
3 |
+
sam2==0.1.0 # Assuming 'sam2' is the package for SAM2 model
|
|
|
|
|
|
|
|
|
|
|
|