upd

app.py

@@ -271,41 +271,8 @@ import time
 
 od_pipe = pipeline("object-detection", "facebook/detr-resnet-50")
 
-"""### tests"""
-
-def test_model_on_image(model, image_path):
-  raw_image = Image.open(image_path)
-  start_time = time.time()
-  pipeline_output = model(raw_image)
-  end_time = time.time()
-  return {"elapsed_time": end_time - start_time, "raw_image": raw_image, "result": pipeline_output}
-
-process_result = test_model_on_image(od_pipe, "sample.jpeg")
-
-process_result
-
-processed_image = render_results_in_image(
-    process_result["raw_image"],
-    process_result["result"])
-
-processed_image
-
-"""## chosen_model ("hustvl/yolos-small"). 123MB"""
-
 chosen_model = pipeline("object-detection", "hustvl/yolos-small")
 
-"""### tests"""
-
-process_result2 = test_model_on_image(chosen_model, "sample.jpeg")
-
-process_result2["result"]
-
-processed_image2 = render_results_in_image(
-    process_result2["raw_image"],
-    process_result2["result"])
-
-processed_image2
-
 """## gradio funcs"""
 
 def get_object_detection_prediction(model_name, raw_image):