Spaces:

SaviAnna
/

History

Runtime error

SaviAnna commited on Jun 9, 2023

Commit

1a8c51c

1 Parent(s): 231519d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ def load_gpt():
         output_attentions = False,
         output_hidden_states = False,
         )
-    model_GPT.load_state_dict(torch.load('model_history.pt', map_location=torch.device('cpu')))
     return model_GPT, tokenizer_GPT
 # # Вешаем сохраненные веса на нашу модель

         output_attentions = False,
         output_hidden_states = False,
         )
+    model_GPT.load_state_dict(torch.load('model_history_friday.pt', map_location=torch.device('cpu')))
     return model_GPT, tokenizer_GPT
 # # Вешаем сохраненные веса на нашу модель