yonigozlan HF Staff commited on
Commit
976c256
·
1 Parent(s): 93e8ba8

change checkpoint

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -13,13 +13,12 @@ from transformers import AutoModelForZeroShotObjectDetection, AutoProcessor
13
 
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
16
- processor = AutoProcessor.from_pretrained("omdet-turbo-tiny-timm")
17
- model = AutoModelForZeroShotObjectDetection.from_pretrained("omdet-turbo-tiny-timm").to(
18
- device
19
- )
20
 
21
  css = """
22
- #warning {background-color: #FFCCCB}
23
  .feedback textarea {font-size: 24px !important}
24
  """
25
 
 
13
 
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
16
+ processor = AutoProcessor.from_pretrained("yonigozlan/omdet-turbo-tiny")
17
+ model = AutoModelForZeroShotObjectDetection.from_pretrained(
18
+ "yonigozlan/omdet-turbo-tiny"
19
+ ).to(device)
20
 
21
  css = """
 
22
  .feedback textarea {font-size: 24px !important}
23
  """
24