Daniel Cerda Escobar
commited on
Commit
·
bd9be39
1
Parent(s):
d3aeb20
Fix model parameters
Browse files
app.py
CHANGED
@@ -135,7 +135,7 @@ if submit:
|
|
135 |
with st.spinner(text="Downloading model weights ... "):
|
136 |
detection_model = get_model(postprocess_match_threshold)
|
137 |
|
138 |
-
slice_size = int(4960/(
|
139 |
image_size = 4960
|
140 |
|
141 |
with st.spinner(text="Performing prediction ... "):
|
|
|
135 |
with st.spinner(text="Downloading model weights ... "):
|
136 |
detection_model = get_model(postprocess_match_threshold)
|
137 |
|
138 |
+
slice_size = int(4960/(slice_number**0.5))
|
139 |
image_size = 4960
|
140 |
|
141 |
with st.spinner(text="Performing prediction ... "):
|