Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ class Gradio:
|
|
56 |
|
57 |
if os.path.isfile(model_path):
|
58 |
model.load_state_dict(
|
59 |
-
torch.load(model_path
|
60 |
)
|
61 |
|
62 |
return model
|
|
|
56 |
|
57 |
if os.path.isfile(model_path):
|
58 |
model.load_state_dict(
|
59 |
+
torch.load(model_path, map_location=self.device), strict=False
|
60 |
)
|
61 |
|
62 |
return model
|