Spaces:

udayjawheri
/

trafficsignclassification

Sleeping

udayjawheri commited on Apr 12, 2024

Commit

5a87ec0

verified ·

1 Parent(s): 81caf3d

Update streamlit_app.py

Files changed (1) hide show

streamlit_app.py CHANGED Viewed

@@ -56,7 +56,7 @@ labels_dict = {
 @st.cache
 def sign_predict(image):
-    model = TFSMLayer("saved_model.pb", call_endpoint='serving_default')
     image = np.array(image, dtype=np.float32)
     image = image/255
     image = np.reshape(image, (1, 32, 32))

 @st.cache
 def sign_predict(image):
+    model = TFSMLayer("./saved_model.pb", call_endpoint='serving_default')
     image = np.array(image, dtype=np.float32)
     image = image/255
     image = np.reshape(image, (1, 32, 32))