safetyWaifu / app.py
Logan Zoellner
debug statements
43e7720
raw
history blame
1.07 kB
from asyncio import constants
import gradio as gr
import requests
import os
import random
def desc_to_image(desc):
random.seed(desc)
tadneSeed=random.randint(0,2**256)
psi=0.7
print("seed",tadneSeed,psi)
#iface = gr.Interface.load("spaces/hysts/TADNE")
#print("about to die",iface,dir(iface))
#img=iface.fns[0].fn(tadneSeed,psi)
print("loading interface")
tadne=gr.Interface.load("spaces/hysts/TADNE")
print("calling interface")
img=tadne(tadneSeed,psi,False)
print("got img",img)
return img
demo = gr.Blocks()
with demo:
gr.Markdown("<h1><center>NPC Generator</center></h1>")
gr.Markdown(
"based on <a href=https://huggingface.co/spaces/hysts/TADNE> TADNE</a>."
)
with gr.Row():
desc_txt = gr.Textbox(label="description",placeholder="0x1f9840a85d5aF5bf1D1762F925BDADdC4201F984")
output_image = gr.Image(label="portrait",type="filepath", shape=(256,256))
b0 = gr.Button("Generate Waifu")
b0.click(desc_to_image,desc_txt,output_image)
#examples=examples
demo.launch(enable_queue=True, debug=True)