| import gradio as gr, os, subprocess, glob, shutil, uuid |
| WORK = "/tmp/bt"; os.makedirs(WORK, exist_ok=True) |
|
|
| def process(img, model): |
| uid = str(uuid.uuid4())[:8] |
| folder = os.path.join(WORK, uid); os.makedirs(folder) |
| inp = os.path.join(folder, "in.png") |
| shutil.copy(img, inp) |
| subprocess.run(f'python main.py --headless --exec_dirs "{folder}" --inpainter {model}', |
| shell=True, cwd="/app", stdout=subprocess.DEVNULL) |
| out = glob.glob(os.path.join(folder, "inpainted", "*")) |
| return out[0] if out else None |
|
|
| gr.Interface( |
| fn=process, |
| inputs=[gr.Image(type="filepath"), gr.Dropdown(["lama_large_512px","lama","aot"], value="lama_large_512px")], |
| outputs=gr.Image(), |
| title="BallonsTranslator Web" |
| ).launch(server_name="0.0.0.0", server_port=7860) |
