Update TheDistanceAssessor.py

TheDistanceAssessor.py

@@ -909,17 +909,4 @@ def run(input_image, model_seg, model_det, image_size, target_distances, num_ys
 
         output_image = get_result(classification, id_map, model.names, borders, image, regions)
         cropped_image = output_image[22:output_image.shape[0] - 40, 74:output_image.shape[1] - 33]
-        return cropped_image
-
-if __name__ == "__main__":
-
-        image_size = [1024,1024]
-        target_distances = [650,1000,2000]
-        num_ys = 10
-
-        PATH_model_seg = 'C:/Users/valac/OneDrive - Západočeská univerzita v Plzni/DP/gradio/SegFormer_B3_1024_finetuned.pth'
-        PATH_model_det = 'C:/Users/valac/OneDrive - Západočeská univerzita v Plzni/DP/gradio/yolov8s.pt'
-        input_image = cv2.imread('C:/Users/valac/OneDrive - Západočeská univerzita v Plzni/DP/gradio/rs00006.jpg') #TO CO VLOZI UZIVATEL
-        model_seg = load_segformer(PATH_model_seg)
-        model_det = load_yolo(PATH_model_det)
-        image = run(input_image, model_seg, model_det, image_size, target_distances, num_ys=num_ys)
+        return cropped_image