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('C:/AI/projects/vision_model_pretrained/Vision_Model_pretrained/models/vision_model_04')
+        accelerator.load_state('/')
         # model = get_model(config, len(tokenizer))
         # model.to(device)