Spaces:

DanielCerda
/

pid_object_detection

Sleeping

Daniel Cerda Escobar commited on Nov 16, 2024

Commit

c1439f5

1 Parent(s): f988731

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 = 4960/(int(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 = int(4960/(float(slice_number)**0.5))
     image_size = 4960
     with st.spinner(text="Performing prediction ... "):