Spaces:

Artgor
/

digit-draw-detect

Running

Andrey commited on Sep 2, 2023

Commit

40a6ccf

•

1 Parent(s): f0facef

Erlemar patch 1 (#18)

* Update requirements.txt

* Update formatting.

* Update formatting.

* Update formatting.

* Update versions

* Change cache to cache_data.

* Change cache to cache_data.

Files changed (1) hide show

src/ml_utils.py +1 -1

src/ml_utils.py CHANGED Viewed

@@ -196,7 +196,7 @@ def predict(
     return nms_boxes
-@st.cache
 def get_model():
     model_name = 'model_files/best_model.pth'

     return nms_boxes
+@st.cache_data
 def get_model():
     model_name = 'model_files/best_model.pth'