handle time consuming requests better

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
-import yolov5
 import sahi.utils
 import sahi.model
 import sahi.predict
+import sahi.slicing
 from PIL import Image
 import numpy
 
@@ -46,6 +46,20 @@ def sahi_yolo_inference(
     postprocess_class_agnostic=False,
 ):
 
+    image_width, image_height = image.size
+    sliced_bboxes = sahi.slicing.get_slice_bboxes(
+        image_height,
+        image_width,
+        slice_height,
+        slice_width,
+        overlap_height_ratio,
+        overlap_width_ratio,
+    )
+    if len(sliced_bboxes) > 60:
+        raise ValueError(
+            f"{len(sliced_bboxes)} slices are too much for huggingface spaces, try smaller slice size."
+        )
+
     # standard inference
     prediction_result_1 = sahi.predict.get_prediction(
         image=image, detection_model=model, image_size=image_size
@@ -125,4 +139,5 @@ gr.Interface(
     article=article,
     examples=examples,
     theme="default",
+    enable_queue=True,
 ).launch(debug=True)