0xMRTT/matrixai
0xMRTT/matrixai
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: service

Browse Source
This commit is contained in:
0xMRTT 2023-06-08 20:25:28 +02:00
parent 8fd7950a62
commit b10685cc06
Signed by: 0xMRTT
GPG Key ID: 910B287304120902
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
matrixai.service
View File

@ -1,7 +1,11 @@
[Unit]
Description=Matrix AI
After=multi-user.target[Service]
After=multi-user.target
[Service]
Type=simple
Restart=always
ExecStart=/usr/bin/python3 /home/0xmrtt/matrixai/run.sh
ExecStart=/home/0xmrtt/matrixai/run.sh
[Install]
WantedBy=multi-user.target

8
matrixai/__init__.py
View File

@ -445,8 +445,12 @@ def run():
async def generate_image(image_prompt, style_value, ratio_value, negative):
imagine = AsyncImagine()
filename = str(uuid.uuid4()) + ".png"
style_enum = Style[style_value]
ratio_enum = Ratio[ratio_value]
try:
style_enum = Style[style_value]
ratio_enum = Ratio[ratio_value]
except KeyError:
style_enum = Style.IMAGINE_V3
ratio_enum = Ratio.RATIO_1X1
img_data = await imagine.sdprem(
prompt=image_prompt,
style=style_enum,