Spaces:
Running
on
Zero
Running
on
Zero
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
|