Spaces:
Sleeping
Sleeping
UPDATE MODES google/paligemma-3b-ft-coco35l-224
Browse files
model.py
CHANGED
|
@@ -6,9 +6,9 @@ from config import MODEL_NAME
|
|
| 6 |
class RadarDetectionModel:
|
| 7 |
def __init__(self):
|
| 8 |
self.feature_extractor = AutoFeatureExtractor.from_pretrained(
|
| 9 |
-
"google/
|
| 10 |
self.model = AutoModelForObjectDetection.from_pretrained(
|
| 11 |
-
"google/
|
| 12 |
self.model.eval()
|
| 13 |
|
| 14 |
@torch.no_grad()
|
|
|
|
| 6 |
class RadarDetectionModel:
|
| 7 |
def __init__(self):
|
| 8 |
self.feature_extractor = AutoFeatureExtractor.from_pretrained(
|
| 9 |
+
"google/paligemma-3b-ft-coco35l-224")
|
| 10 |
self.model = AutoModelForObjectDetection.from_pretrained(
|
| 11 |
+
"google/paligemma-3b-ft-coco35l-224")
|
| 12 |
self.model.eval()
|
| 13 |
|
| 14 |
@torch.no_grad()
|