123LETSPLAY commited on
Commit
0e9a7e8
1 Parent(s): 20f29bf

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -9
requirements.txt CHANGED
@@ -1,9 +1,3 @@
1
- # Use SAM2 with images
2
- import torch
3
- from sam2.sam2_image_predictor import SAM2ImagePredictor
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