Spaces:

austindavis
/

play-chess-gpt2-uci-8x8x512

Sleeping

austindavis commited on May 24

Commit

44e120c

•

1 Parent(s): 1c8df7b

Update app.py

Updated model to austindavis/chess-gpt2-uci-8x8x512

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import numpy as np
 import io
 import traceback
 # %%
-checkpoint_name = "austindavis/gpt2-lichess-uci-201601"
 tokenizer = UciTileTokenizer()
 model: GPT2LMHeadModel = AutoModelForCausalLM.from_pretrained(checkpoint_name)
 model.requires_grad_(False)

 import io
 import traceback
 # %%
+checkpoint_name = "austindavis/chess-gpt2-uci-8x8x512"
 tokenizer = UciTileTokenizer()
 model: GPT2LMHeadModel = AutoModelForCausalLM.from_pretrained(checkpoint_name)
 model.requires_grad_(False)