Spaces:
Sleeping
Sleeping
Samanta Das
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from yolo import FractureDetector
|
|
14 |
from llama import generate_response_based_on_yolo
|
15 |
|
16 |
# Define model path and output folder relative to the app's directory
|
17 |
-
MODEL_PATH = '
|
18 |
OUTPUT_FOLDER = 'output_images' # Use a relative path to avoid permission issues
|
19 |
|
20 |
# Ensure the output folder exists
|
|
|
14 |
from llama import generate_response_based_on_yolo
|
15 |
|
16 |
# Define model path and output folder relative to the app's directory
|
17 |
+
MODEL_PATH = 'yolov8n_custom_exported.pt' # Update this to the relative path of your model on Hugging Face Spaces
|
18 |
OUTPUT_FOLDER = 'output_images' # Use a relative path to avoid permission issues
|
19 |
|
20 |
# Ensure the output folder exists
|