Spaces:

jackrui
/

Diff-AMP

Sleeping

jackrui commited on Oct 10, 2023

Commit

e23981a

1 Parent(s): 2f1b920

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ def classify_sequence(sequence):
 # 加载模型
 model = MyModel()
-model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')))
 if __name__ == "__main__":

 # 加载模型
 model = MyModel()
+model.load_state_dict(torch.load("best_model.pth"))
 if __name__ == "__main__":