Upload model.py with huggingface_hub

model.py

@@ -452,17 +452,18 @@ class BT4(nn.Module):
 
         return policy_logits, value_winner, value_q
     
-    def get_move_from_fen_no_thinking(self, fen_or_moves: Union[str, List[str]], T: float, device: str = None, **kwargs) -> str:
+    def get_move_from_history(self, fen_or_moves: Union[str, List[str]], T: float, device: str = None, **kwargs) -> str:
         """
-        Predict a move from a FEN position or move history without thinking/search.
+        Predict a move from a move history or FEN position.
         
         Args:
             fen_or_moves: Either a FEN string representing the chess position, or a list of UCI moves
             T: Temperature for sampling (0.0 = deterministic/argmax, >0.0 = stochastic)
             device: Device to run the model on (if None, uses model's device)
+            return_probs: If True, returns a dictionary of move probabilities instead of a single move
         
         Returns:
-            UCI move string (e.g., 'e2e4')
+            UCI move string (e.g., 'e2e4') or dictionary of move probabilities if return_probs=True
         """
         # Detect device from model if not provided
         if device is None: