Spaces:
Sleeping
Sleeping
Fix model loading path - use root directory
Browse files
app.py
CHANGED
|
@@ -32,8 +32,8 @@ import gradio as gr
|
|
| 32 |
|
| 33 |
# ========== CONFIGURATION ==========
|
| 34 |
|
| 35 |
-
# Model directory
|
| 36 |
-
MODEL_DIR = os.path.
|
| 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 = [
|