Update app.py
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ def classify_sequence(sequence):
|
|
| 77 |
|
| 78 |
# 加载模型
|
| 79 |
model = MyModel()
|
| 80 |
-
model.load_state_dict(torch.load("best_model.pth"))
|
| 81 |
|
| 82 |
|
| 83 |
if __name__ == "__main__":
|
|
|
|
| 77 |
|
| 78 |
# 加载模型
|
| 79 |
model = MyModel()
|
| 80 |
+
model.load_state_dict(torch.load("best_model.pth"), strict=False)
|
| 81 |
|
| 82 |
|
| 83 |
if __name__ == "__main__":
|