CineAI commited on
Commit
3d49ae1
·
verified ·
1 Parent(s): 5d676fa

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -5
src/streamlit_app.py CHANGED
@@ -71,11 +71,12 @@ def load_model(model_path):
71
  st.error(f"Error loading model: {e}")
72
  return None
73
 
74
- MODEL_PATH = get_file(
75
- force_download=False,
76
- origin="https://huggingface.co/CineAI/RSSI/resolve/main/rssi_last.pt",
77
- file_hash="c689b8a006e574bf5e4f2ff252230cb59ac3de42ca24875000144c2f552bd837"
78
- )
 
79
  print(f"MODEL_PATH: {MODEL_PATH}")
80
  model = load_model(MODEL_PATH)
81
 
 
71
  st.error(f"Error loading model: {e}")
72
  return None
73
 
74
+ # MODEL_PATH = get_file(
75
+ # force_download=False,
76
+ # origin="https://huggingface.co/CineAI/RSSI/resolve/main/rssi_last.pt",
77
+ # file_hash="c689b8a006e574bf5e4f2ff252230cb59ac3de42ca24875000144c2f552bd837"
78
+ # )
79
+ MODEL_PATH = './src/rssi_last.pt'
80
  print(f"MODEL_PATH: {MODEL_PATH}")
81
  model = load_model(MODEL_PATH)
82