Commit
·
971afa1
1
Parent(s):
f9f05d9
correctly named hubert pth
Browse files
app.py
CHANGED
|
@@ -246,7 +246,7 @@ associated_links = {}
|
|
| 246 |
def load_hubert():
|
| 247 |
global hubert_model
|
| 248 |
# Load the model
|
| 249 |
-
hubert_model = torch.load("
|
| 250 |
|
| 251 |
# Prepare the model
|
| 252 |
hubert_model = hubert_model.to(config.device)
|
|
|
|
| 246 |
def load_hubert():
|
| 247 |
global hubert_model
|
| 248 |
# Load the model
|
| 249 |
+
hubert_model = torch.load("hubert.pth", map_location=config.device)
|
| 250 |
|
| 251 |
# Prepare the model
|
| 252 |
hubert_model = hubert_model.to(config.device)
|