Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from PIL import Image, ImageDraw
-import sys
 import os
 import time
 import threading
@@ -15,13 +14,13 @@ from llama import generate_response_based_on_yolo
 
 # Define model path and output folder relative to the app's directory
 MODEL_PATH = 'yolov8n_custom.pkl'  # Update this to the relative path of your model on Hugging Face Spaces
-OUTPUT_FOLDER = '/app/output_images'  # Default output folder in Hugging Face Spaces
+OUTPUT_FOLDER = './output_images'  # Use a relative path to avoid permission issues
 
 # Ensure the output folder exists
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 
 # Initialize the fracture detector
-detector = FractureDetector(MODEL_PATH, OUTPUT_FOLDER)
+detector = FractureDetector(MODEL_PATH)
 
 def delete_file_after_delay(file_path, delay):
     """Delete the specified file after a given delay."""