Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -380,10 +380,8 @@ depth_estimator = depth_estimator.to(device)
|
|
380 |
# ---------------------------------------
|
381 |
detr_model = DetrForObjectDetection.from_pretrained("facebook-detr-50",
|
382 |
# id2label={v:k for k,v in CLASS2ID.items()},
|
383 |
-
dtype=dtype,
|
384 |
use_safetensors=True)
|
385 |
detr_processor = DetrImageProcessor.from_pretrained("facebook-detr-50",
|
386 |
-
dtype=dtype,
|
387 |
use_safetensors=True)
|
388 |
|
389 |
def main():
|
|
|
380 |
# ---------------------------------------
|
381 |
detr_model = DetrForObjectDetection.from_pretrained("facebook-detr-50",
|
382 |
# id2label={v:k for k,v in CLASS2ID.items()},
|
|
|
383 |
use_safetensors=True)
|
384 |
detr_processor = DetrImageProcessor.from_pretrained("facebook-detr-50",
|
|
|
385 |
use_safetensors=True)
|
386 |
|
387 |
def main():
|