WarBot / WarServer.py
kertser's picture
Upload WarServer.py
17e09e4
raw history blame
No virus
827 Bytes
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 = WarBot.get_response(received_string, model, tokenizer, model_punct)
response_string = response
conn.sendall(response_string.encode())
conn.close()