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