Spaces:

DanielCerda
/

pid_object_detection

Sleeping

Daniel Cerda Escobar commited on Nov 16, 2024

Commit

f988731

1 Parent(s): 5404556

Fix model parameters

Files changed (1) hide show

app.py CHANGED Viewed

@@ -138,7 +138,7 @@ if submit:
     with st.spinner(text="Downloading model weights ... "):
         detection_model = get_model(postprocess_match_threshold)
-    slice_size = int(4960/(slice_number**0.5))
     image_size = 4960
     with st.spinner(text="Performing prediction ... "):

     with st.spinner(text="Downloading model weights ... "):
         detection_model = get_model(postprocess_match_threshold)
+    slice_size = 4960/(int(slice_number)**0.5)
     image_size = 4960
     with st.spinner(text="Performing prediction ... "):