Hugging Face's logo

Spaces:
siriuszeina
/
Interpolation-Frame-Animagine-XL-3.1
Running

App Files Community
Interpolation-Frame-Animagine-XL-3.1 / app.py
siriuszeina's picture
siriuszeina
Update app.py
b9f80c9 verified
raw history blame contribute delete
No virus
6.41 kB
import streamlit as st
import shutil
import os
from PIL import Image
import requests
from gradio_client import Client
prompt_prefix = "night, full body, closed mouth, happy face, walking front, "
prompt_list = [
"loli, onepiece dress, white dress, sleeveless",
"loli, onepiece dress, white dress, sleeveless, white belt",
"loli, onepiece dress, white dress, sleeveless, white belt, white ribbon",
"loli, onepiece dress, white dress, sleeveless, white belt, white ribbon, white sock",
"loli, onepiece dress, white dress, sleeveless, white belt, white ribbon, white sock, white shoes",
"loli, onepiece dress, white dress, sleeveless, black belt, white ribbon, white sock, white shoes",
"loli, onepiece dress, white dress, sleeveless, black belt, red ribbon, white sock, white shoes",
"loli, onepiece dress, white dress, sleeveless, black belt, red ribbon, white sock, black shoes",
"loli, onepiece dress, white dress, sleeveless, black belt, red ribbon, white sock, black shoes, white skirt",
"loli, onepiece dress, white dress, short sleeve, black belt, red ribbon, white sock, black shoes, white skirt",
"loli, onepiece dress, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, white skirt",
"loli, elementary school dress, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, white skirt",
"loli, elementary school uniform, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, white skirt",
"loli, elementary school uniform, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, white skirt, white collar",
"loli, elementary school uniform, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, white skirt, white collar, sailor collar",
"loli, elementary school uniform, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, blue skirt, white collar, sailor collar",
"loli, middle school uniform, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, blue skirt, white collar, sailor collar",
"loli, middle school uniform, white clothes, short sleeve, black belt, red ribbon, white sock, black shoes, blue skirt, blue collar, sailor collar",
"loli, middle school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, blue collar, sailor collar",
"loli, middle school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, white collar, sailor collar",
"loli, middle school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, white collar, formal collar",
"middle school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, white collar, formal collar",
"high school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, white collar, formal collar",
"high school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, white collar, formal collar, blazer",
"high school uniform, white clothes, short sleeve, black belt, red necktie, white sock, black shoes, blue skirt, white collar, formal collar, black blazer",
]
prompt_suffix = ", masterpiece, best quality, very aesthetic, absurdres"
if os.path.exists("./output/") == False:
os.mkdir("./output/")
if os.path.exists("./frames/") == False:
os.mkdir("./frames/")
if os.path.exists("./interpolations/") == False:
os.mkdir("./interpolations/")
project_name = str(st.text_input("Project name")).replace(" ", "-")
character = st.text_input("character")
anime_from = st.text_input("From")
seed = st.number_input('seed')
if st.button("Generate!"):
client = Client("cagliostrolab/animagine-xl-3.1")
n = 0
for p in prompt_list:
prompt_new = "1girl, "+character+", "+anime_from+", "+prompt_prefix+p+prompt_suffix
result = client.predict(
prompt_new, # str in 'Prompt' Textbox component
"nsfw, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]", # str in 'Negative Prompt' Textbox component
seed, # float (numeric value between 0 and 2147483647) in 'Seed' Slider component
768, # float (numeric value between 512 and 2048) in 'Width' Slider component
1344, # float (numeric value between 512 and 2048) in 'Height' Slider component
7, # float (numeric value between 1 and 12) in 'Guidance scale' Slider component
28, # float (numeric value between 1 and 50) in 'Number of inference steps' Slider component
"Euler a", # Literal['DPM++ 2M Karras', 'DPM++ SDE Karras', 'DPM++ 2M SDE Karras', 'Euler', 'Euler a', 'DDIM'] in 'Sampler' Dropdown component
"768 x 1344", # Literal['1024 x 1024', '1152 x 896', '896 x 1152', '1216 x 832', '832 x 1216', '1344 x 768', '768 x 1344', '1536 x 640', '640 x 1536', 'Custom'] in 'Aspect Ratio' Radio component
"(None)", # Literal['(None)', 'Cinematic', 'Photographic', 'Anime', 'Manga', 'Digital Art', 'Pixel art', 'Fantasy art', 'Neonpunk', '3D Model'] in 'Style Preset' Radio component
"Standard v3.1", # Literal['(None)', 'Standard v3.0', 'Standard v3.1', 'Light v3.1', 'Heavy v3.1'] in 'Quality Tags Presets' Dropdown component
False, # bool in 'Use Upscaler' Checkbox component
0, # float (numeric value between 0 and 1) in 'Strength' Slider component
1, # float (numeric value between 1 and 1.5) in 'Upscale by' Slider component
True, # bool in 'Add Quality Tags' Checkbox component
api_name="/run"
)
source = 'https://cagliostrolab-animagine-xl-3-1.hf.space/file='+result[0][0]["image"]
frame = Image.open(requests.get(source, stream=True).raw)
frame.save("./output/"+str(n)+".png")
n += 1
print(source)
output_num = len(os.listdir("./output/"))
if output_num > 0:
archived = shutil.make_archive("output", 'zip', "./output/")
st.download_button(label="Download "+str(output_num)+" file", data=archived, file_name='output.zip', mime='application/x-zip')
print(os.listdir("./"))