Spaces:
Runtime error
Runtime error
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()
|