| |
| import socket |
| import WarBot |
|
|
| import warnings |
| warnings.filterwarnings("ignore") |
|
|
| model,tokenizer,model_punct = WarBot.initialize() |
|
|
| HOST = '10.0.0.125' |
| PORT = 5000 |
|
|
| with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket: |
| server_socket.bind((HOST, PORT)) |
| server_socket.listen() |
| print(f'Server is listening on port {PORT}') |
| while True: |
| conn, addr = server_socket.accept() |
| with conn: |
| print(f'Connected by {addr}') |
| data = conn.recv(1024) |
| received_string = data.decode() |
| print(f'Received string from client: {received_string}') |
|
|
| response = "" |
| while not response: |
| response = WarBot.get_response(received_string, model, tokenizer, model_punct, temperature=0.2) |
| response_string = response |
|
|
| conn.sendall(response_string.encode()) |
| conn.close() |
