Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
7a4aa57
1
Parent(s):
043100e
update weights path
Browse files
app.py
CHANGED
@@ -13,9 +13,9 @@ from transformers import AutoModelForZeroShotObjectDetection, AutoProcessor
|
|
13 |
|
14 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
15 |
|
16 |
-
processor = AutoProcessor.from_pretrained("
|
17 |
model = AutoModelForZeroShotObjectDetection.from_pretrained(
|
18 |
-
"
|
19 |
).to(device)
|
20 |
|
21 |
css = """
|
|
|
13 |
|
14 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
15 |
|
16 |
+
processor = AutoProcessor.from_pretrained("omlab/omdet-turbo-swin-tiny-hf")
|
17 |
model = AutoModelForZeroShotObjectDetection.from_pretrained(
|
18 |
+
"omlab/omdet-turbo-swin-tiny-hf"
|
19 |
).to(device)
|
20 |
|
21 |
css = """
|