Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
import os
|
2 |
-
import gradio as gr
|
3 |
import random
|
|
|
|
|
|
|
4 |
import chess
|
5 |
import chess.svg
|
6 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
@@ -44,6 +46,7 @@ def predict_move(fen, top_k=3):
|
|
44 |
return p['label']
|
45 |
|
46 |
def btn_load(inp_fen):
|
|
|
47 |
board = chess.Board()
|
48 |
|
49 |
with open('board.svg', 'w') as f:
|
@@ -51,6 +54,7 @@ def btn_load(inp_fen):
|
|
51 |
return 'board.svg', board.fen(), ''
|
52 |
|
53 |
def btn_play(inp_fen, inp_move, inp_notation, inp_k):
|
|
|
54 |
board = chess.Board(inp_fen)
|
55 |
|
56 |
if inp_move:
|
|
|
1 |
import os
|
|
|
2 |
import random
|
3 |
+
from datetime import datetime
|
4 |
+
|
5 |
+
import gradio as gr
|
6 |
import chess
|
7 |
import chess.svg
|
8 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
|
|
46 |
return p['label']
|
47 |
|
48 |
def btn_load(inp_fen):
|
49 |
+
print(f'** log - load - ts {datetime.now().isoformat()}, fen: {inp_fen}')
|
50 |
board = chess.Board()
|
51 |
|
52 |
with open('board.svg', 'w') as f:
|
|
|
54 |
return 'board.svg', board.fen(), ''
|
55 |
|
56 |
def btn_play(inp_fen, inp_move, inp_notation, inp_k):
|
57 |
+
print(f'** log - play - ts {datetime.now().isoformat()}, fen: {inp_fen}, movw: {inp_move}, notation: {inp_notation}, top_k: {inp_k}')
|
58 |
board = chess.Board(inp_fen)
|
59 |
|
60 |
if inp_move:
|