Spaces:

Lwasinam
/

image-caption

Sleeping

Lwasinam commited on Mar 27, 2024

Commit

ce77695

verified ·

1 Parent(s): 63dd1a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -144,7 +144,7 @@ def start():
     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')
     image = image_base64()

     tokenizer = get_or_build_tokenizer(config)
     model = get_model(config, len(tokenizer))
     model = accelerator.prepare(model)
+    accelerator.load_state('/')
     image = image_base64()