Update app.py

app.py

@@ -170,7 +170,7 @@ def main():
         tokenizer = get_or_build_tokenizer(config)
         model = get_model(config, len(tokenizer))
         model = accelerator.prepare(model)
-        accelerator.load_state('')
+        accelerator.load_state('models/')
         # model = get_model(config, len(tokenizer))
         # model.to(device)