skyBluezz commited on
Commit
280e23e
·
verified ·
1 Parent(s): 1717c15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
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():