tenet commited on
Commit
146ff53
·
verified ·
1 Parent(s): b6b5359

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from main import run_mcts_cot
3
+
4
+ # Create the Gradio interface
5
+ initial_board = [[0, 0, 0], [0, 0, 0], [0, 0, 0]]
6
+ iface = gr.Interface(
7
+ fn=run_mcts_cot,
8
+ inputs=gr.inputs.JSON(),
9
+ outputs=["text", "text"],
10
+ title="RWKV CoT Demo for MCTS",
11
+ description="This demo uses RWKV to generate Chain-of-Thought reasoning to guide the MCTS algorithm in a Tic-Tac-Toe game."
12
+ )
13
+
14
+ # Launch the interface
15
+ if __name__ == "__main__":
16
+ iface.launch()