Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ MAP_TILES = {
|
|
| 28 |
}
|
| 29 |
|
| 30 |
# Model configuration - corrected model name
|
| 31 |
-
MODEL_NAME = "numind/NuExtract-
|
| 32 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 33 |
TORCH_DTYPE = torch.bfloat16 if DEVICE == "cuda" else torch.float32
|
| 34 |
MAX_INPUT_LENGTH = 20000 # For sliding window processing
|
|
|
|
| 28 |
}
|
| 29 |
|
| 30 |
# Model configuration - corrected model name
|
| 31 |
+
MODEL_NAME = "numind/NuExtract-1.5" # Fixed model name according to documentation
|
| 32 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 33 |
TORCH_DTYPE = torch.bfloat16 if DEVICE == "cuda" else torch.float32
|
| 34 |
MAX_INPUT_LENGTH = 20000 # For sliding window processing
|