Spaces:

danielle2003
/

ca

Sleeping

danielle2003 commited on Jan 20

Commit

d826e76

verified ·

1 Parent(s): 8696997

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from torchvision import transforms
 import torch.nn.functional as F
 if "model" not in st.session_state:
-    st.session_state.model = None
 if "choice" not in st.session_state:
     st.session_state.choice = "tensorflow"

 import torch.nn.functional as F
 if "model" not in st.session_state:
+    st.session_state.model = tf.keras.models.load_model('best_model.keras')
 if "choice" not in st.session_state:
     st.session_state.choice = "tensorflow"