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

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -0
src/streamlit_app.py CHANGED
@@ -76,6 +76,7 @@ MODEL_PATH = get_file(
76
  origin="https://huggingface.co/CineAI/RSSI/resolve/main/rssi_last.pt",
77
  file_hash="c689b8a006e574bf5e4f2ff252230cb59ac3de42ca24875000144c2f552bd837"
78
  )
 
79
  model = load_model(MODEL_PATH)
80
 
81
  # --- Sidebar ---
@@ -100,6 +101,7 @@ st.title("🖼️ Custom Object Detection with YOLO")
100
  st.write("Upload an image via the sidebar to see the model's predictions.")
101
 
102
  if uploaded_file is not None:
 
103
  # Read the uploaded image file
104
  image_data = uploaded_file.getvalue()
105
  original_image = Image.open(io.BytesIO(image_data))
 
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
 
82
  # --- Sidebar ---
 
101
  st.write("Upload an image via the sidebar to see the model's predictions.")
102
 
103
  if uploaded_file is not None:
104
+ print(f"uploaded_file: {uploaded_file}")
105
  # Read the uploaded image file
106
  image_data = uploaded_file.getvalue()
107
  original_image = Image.open(io.BytesIO(image_data))