Update src/streamlit_app.py
Browse files- 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))
|