Spaces:
Sleeping
Sleeping
Commit ·
11672f5
1
Parent(s): 4baea7b
fix: edit streamlit
Browse files- best.pt → src/best.pt +0 -0
- src/streamlit_app.py +3 -1
best.pt → src/best.pt
RENAMED
|
File without changes
|
src/streamlit_app.py
CHANGED
|
@@ -2,9 +2,11 @@ import streamlit as st
|
|
| 2 |
from ultralytics import YOLO
|
| 3 |
from PIL import Image
|
| 4 |
import numpy as np
|
|
|
|
| 5 |
|
| 6 |
# --- CONFIGURATION ---
|
| 7 |
-
|
|
|
|
| 8 |
|
| 9 |
# --- KNOWLEDGE BASE ---
|
| 10 |
SYMPTOM_QUESTIONS = {
|
|
|
|
| 2 |
from ultralytics import YOLO
|
| 3 |
from PIL import Image
|
| 4 |
import numpy as np
|
| 5 |
+
import os
|
| 6 |
|
| 7 |
# --- CONFIGURATION ---
|
| 8 |
+
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 9 |
+
MODEL_PATH = os.path.join(SCRIPT_DIR, "best.pt")
|
| 10 |
|
| 11 |
# --- KNOWLEDGE BASE ---
|
| 12 |
SYMPTOM_QUESTIONS = {
|