Montey/python-edge
0
1o
2 AS!gn � @ s� d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z m3Z4 d dl m Z d dl
Z
d dlmZ d dlmZ d d lmZmZ e e e�jjd5 �Zd d � Zd
d� ZedkrYe� dS dS )� N)�BytesIO)�Listener)�chmod�remove)�abspath�exists)�Path)�
JpegImageFile)�TextToImageRequest)�
load_pipeline�inferzinferences.sockc C s t j�� d S )N)�torch�cuda� empty_cache� r r �/sandbox/src/main.py�at_exit s r c 6 C s t �t� td� t� } tdt� d�� tt�r tt� tt��e}t td� td� |�7� �G}td� z t � |�
� �d��}W n tyZ td � Y W d � W d � d S w t|| �}t� }|j|tjd8� |�� }|�|� q41 sxw Y W d � d S 1 s�w Y d S ) NzLoading pipelinez%Pipeline loaded, creating socket at '�'i� zAwaiting connections� ConnectedTzutf-8zInference socket exiting)�format)�atexit�registerr �printr �SOCKETr r r r �acceptr9 �model_validate_json�10recv_bytes�decode�EOFErrorr r �saver r �getvalue�11send_bytes)�pipeline�listener�12connection�request�image�data�packetr r r �main s8 13141516 ��17 1819��"�r) �__main__)r �ior �multiprocessing.connectionr �osr r �os.pathr r �pathlibr r
�PIL.JpegImagePluginr �pipelines.modelsr20 r" r r �__file__�parentr r r) �__name__r r r r �<module> s $21�