polejowska commited on
Commit
a1b8785
1 Parent(s): 340429d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -41,6 +41,7 @@ def detect_objects(model_name, image_input, threshold):
41
  model_details = "DETR details"
42
  elif "YOLOS" in model_name:
43
  model = YolosForObjectDetection.from_pretrained(MODELS_REPO[model_name])
 
44
 
45
  (
46
  processed_outputs,
 
41
  model_details = "DETR details"
42
  elif "YOLOS" in model_name:
43
  model = YolosForObjectDetection.from_pretrained(MODELS_REPO[model_name])
44
+ model_details = "YOLOS details"
45
 
46
  (
47
  processed_outputs,