Spaces:
Runtime error
Runtime error
Commit ·
fc4bdb9
1
Parent(s): 3b01972
Fix invalid model parameters
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ import torch
|
|
| 31 |
print("Preloading Allosaurus model...")
|
| 32 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 33 |
print(f"Using device: {device}")
|
| 34 |
-
MODEL = read_recognizer(alt_model_path="/tmp/allosaurus_models"
|
| 35 |
if device == 'cuda':
|
| 36 |
MODEL.model.to(device)
|
| 37 |
|
|
|
|
| 31 |
print("Preloading Allosaurus model...")
|
| 32 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 33 |
print(f"Using device: {device}")
|
| 34 |
+
MODEL = read_recognizer(alt_model_path="/tmp/allosaurus_models")
|
| 35 |
if device == 'cuda':
|
| 36 |
MODEL.model.to(device)
|
| 37 |
|