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

provider: add anything

Browse Source
This commit is contained in:
0xMRTT 2023-05-06 17:51:41 +02:00
parent c6dbc3812f
commit f9b096e624
Signed by: 0xMRTT
GPG Key ID: 19C1449A774028BD
2 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main.py
View File

@ -161,12 +161,14 @@ class ImaginerApplication(Adw.Application):
NITRO_DIFFUSION = 4 NITRO_DIFFUSION = 4
ANALOG_DIFFUSION = 5 ANALOG_DIFFUSION = 5
PORTRAIT_PLUS = 6 PORTRAIT_PLUS = 6
ANYTHING = 7
prompt = self.win.prompt.get_text() prompt = self.win.prompt.get_text()
negative_prompt = self.win.negative_prompt.get_text() negative_prompt = self.win.negative_prompt.get_text()
self.token = self.win.token.get_text() self.token = self.win.token.get_text()
openai.api_key = self.token openai.api_key = self.token
HUGGINGFACE_API = "https://api-inference.huggingface.co/models"
def thread_run(): def thread_run():
try: try:
@ -196,7 +198,7 @@ class ImaginerApplication(Adw.Application):
"inputs": prompt, "inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "", "negative_prompts": negative_prompt if negative_prompt else "",
}, },
"https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1", f"{HUGGINGFACE_API}/stabilityai/stable-diffusion-2-1",
) )
path = f"{path}-stable-diffusion.png" path = f"{path}-stable-diffusion.png"
case ProvidersEnum.WAIFU_DIFFUSION.value: case ProvidersEnum.WAIFU_DIFFUSION.value:
@ -205,7 +207,7 @@ class ImaginerApplication(Adw.Application):
"inputs": prompt, "inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "", "negative_prompts": negative_prompt if negative_prompt else "",
}, },
"https://api-inference.huggingface.co/models/hakurei/waifu-diffusion", f"{HUGGINGFACE_API}/hakurei/waifu-diffusion",
) )
path = f"{path}-waifu-diffusion.png" path = f"{path}-waifu-diffusion.png"
case ProvidersEnum.OPENJOURNEY.value: case ProvidersEnum.OPENJOURNEY.value:
@ -214,7 +216,7 @@ class ImaginerApplication(Adw.Application):
"inputs": prompt, "inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "", "negative_prompts": negative_prompt if negative_prompt else "",
}, },
"https://api-inference.huggingface.co/models/prompthero/openjourney-v4", f"{HUGGINGFACE_API}/prompthero/openjourney-v4",
) )
path = f"{path}-openjourney.png" path = f"{path}-openjourney.png"
case ProvidersEnum.NITRO_DIFFUSION.value: case ProvidersEnum.NITRO_DIFFUSION.value:
@ -223,7 +225,7 @@ class ImaginerApplication(Adw.Application):
"inputs": prompt, "inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "", "negative_prompts": negative_prompt if negative_prompt else "",
}, },
"https://api-inference.huggingface.co/models/nitrosocke/Nitro-Diffusion", f"{HUGGINGFACE_API}/nitrosocke/Nitro-Diffusion",
) )
path = f"{path}-nitro-diffusion.png" path = f"{path}-nitro-diffusion.png"
case ProvidersEnum.ANALOG_DIFFUSION.value: case ProvidersEnum.ANALOG_DIFFUSION.value:
@ -232,7 +234,7 @@ class ImaginerApplication(Adw.Application):
"inputs": prompt, "inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "", "negative_prompts": negative_prompt if negative_prompt else "",
}, },
"https://api-inference.huggingface.co/models/wavymulder/Analog-Diffusion", f"{HUGGINGFACE_API}/wavymulder/Analog-Diffusion",
) )
path = f"{path}-analog-diffusion.png" path = f"{path}-analog-diffusion.png"
case ProvidersEnum.PORTRAIT_PLUS.value: case ProvidersEnum.PORTRAIT_PLUS.value:
@ -241,9 +243,18 @@ class ImaginerApplication(Adw.Application):
"inputs": prompt, "inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "", "negative_prompts": negative_prompt if negative_prompt else "",
}, },
"https://api-inference.huggingface.co/models/wavymulder/portraitplus", f"{HUGGINGFACE_API}/wavymulder/portraitplus",
) )
path = f"{path}-portrait-plus.png" path = f"{path}-portrait-plus.png"
case ProvidersEnum.ANYTHING.value:
image_bytes = self.query(
{
"inputs": prompt,
"negative_prompts": negative_prompt if negative_prompt else "",
},
f"{HUGGINGFACE_API}/andite/anything-v4.0",
)
path = f"{path}-anything.png"
if image_bytes: if image_bytes:
try: try:
image = Image.open(io.BytesIO(image_bytes)) image = Image.open(io.BytesIO(image_bytes))

2
src/window.blp
View File

@ -149,5 +149,5 @@ menu main_menu {
} }
Gtk.StringList provider_model{ Gtk.StringList provider_model{
strings [_("Stable Diffusion"), _("OpenAI"), _("Waifu Diffusion"), _("Open Journey"), _("Nitro Diffusion"), _("Analog Diffusion"), _("Portrait Plus")] strings [_("Stable Diffusion"), _("OpenAI"), _("Waifu Diffusion"), _("Open Journey"), _("Nitro Diffusion"), _("Analog Diffusion"), _("Portrait Plus"), _("Anything")]
} }