Update app.py

app.py

@@ -1,46 +1,17 @@
-def print_board(board):
-    for row in board:
-        print(" | ".join(row))
-        print("-" * 9)
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-def check_winner(board, player):
-    # Check rows, columns, and diagonals
-    for i in range(3):
-        if all(cell == player for cell in board[i]) or all(board[j][i] == player for j in range(3)):
-            return True
-    if all(board[i][i] == player for i in range(3)) or all(board[i][2 - i] == player for i in range(3)):
-        return True
-    return False
+model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
 
-def check_tie(board):
-    return all(cell != " " for row in board for cell in row)
+model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
 
-def tic_tac_toe():
-    board = [[" " for _ in range(3)] for _ in range(3)]
-    current_player = "X"
+messages = [
+    {"role": "user", "content": "What is your favourite condiment?"},
+    {"role": "assistant", "content": "Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!"},
+    {"role": "user", "content": "Do you have mayonnaise recipes?"}
+]
 
-    print("Welcome to Tic-Tac-Toe!")
+inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to("cuda")
 
-    while True:
-        print_board(board)
-
-        row = int(input(f"Player {current_player}, enter row (0, 1, or 2): "))
-        col = int(input(f"Player {current_player}, enter column (0, 1, or 2): "))
-
-        if board[row][col] == " ":
-            board[row][col] = current_player
-            if check_winner(board, current_player):
-                print_board(board)
-                print(f"Player {current_player} wins! Congratulations!")
-                break
-            elif check_tie(board):
-                print_board(board)
-                print("It's a tie! Well played.")
-                break
-            else:
-                current_player = "O" if current_player == "X" else "X"
-        else:
-            print("Cell already occupied. Try again.")
-
-if __name__ == "__main__":
-    tic_tac_toe()
+outputs = model.generate(inputs, max_new_tokens=20)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))