Spaces:

muhammadsalmanalfaridzi
/

testing-roboflow

Running

muhammadsalmanalfaridzi commited on Jan 9

Commit

21a3b22

verified ·

1 Parent(s): 25430ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ def detect_objects(image):
     predictions = get_sliced_prediction(
         image_path=temp_file_path,
         model_path="path_to_your_model",
-        model_type="yolov5",  # Adjust based on your model
         slice_height=256,
         slice_width=256,
         overlap_height_ratio=0.1,

     predictions = get_sliced_prediction(
         image_path=temp_file_path,
         model_path="path_to_your_model",
+        model_type="yolov8",  # Adjust based on your model
         slice_height=256,
         slice_width=256,
         overlap_height_ratio=0.1,