Samanta Das commited on
Commit
a07e44f
·
verified ·
1 Parent(s): dfaa54a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 'yolov8n_custom.pkl' # 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
 
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