Spaces:
Runtime error
Runtime error
Commit
·
976c256
1
Parent(s):
93e8ba8
change checkpoint
Browse files
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
|
| 17 |
-
model = AutoModelForZeroShotObjectDetection.from_pretrained(
|
| 18 |
-
|
| 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 |
|