Meet2304 commited on
Commit
b0364d3
·
1 Parent(s): d87f7c0

Fix model loading path - use root directory

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ import gradio as gr
32
 
33
  # ========== CONFIGURATION ==========
34
 
35
- # Model directory (local path relative to this file)
36
- MODEL_DIR = os.path.join(os.path.dirname(__file__), "convnextv2_cervical_v0.1")
37
 
38
  # Class names
39
  CLASS_NAMES = [
 
32
 
33
  # ========== CONFIGURATION ==========
34
 
35
+ # Model directory - model files are in the root directory of the Space
36
+ MODEL_DIR = os.path.dirname(__file__) # Current directory where app.py is located
37
 
38
  # Class names
39
  CLASS_NAMES = [