pc4work commited on
Commit
fd694bc
·
verified ·
1 Parent(s): 793e1cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -1,8 +1,14 @@
1
- from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
2
- import torch
3
- from transformers import SamModel, SamProcessor
4
  import spaces
 
5
  import numpy as np
 
 
 
 
 
 
 
 
6
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
7
 
8
  sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
 
 
 
 
1
  import spaces
2
+ import torch
3
  import numpy as np
4
+
5
+ from transformers import (
6
+ AutoProcessor,
7
+ AutoModelForZeroShotObjectDetection,
8
+ SamModel,
9
+ SamProcessor,
10
+ )
11
+
12
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
13
 
14
  sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")