img2img / app.py
Furkan12's picture
d
3631e5f verified
raw
history blame
680 Bytes
import gradio as gr
# Modeli yükle
model = gr.load("models/radames/stable-diffusion-2-1-img2img")
def my_custom_function(image, prompt):
# Modeli doğrudan çağır
output = model(image=image, prompt=prompt)
# Burada output üzerinde ek işlemler yapabilirsiniz
# Örneğin, çıktıyı yeniden boyutlandırma, filtreleme vb.
return output
iface = gr.Interface(
fn=my_custom_function,
inputs=[gr.Image(type="pil"), gr.Textbox()],
outputs=gr.Image(type="pil"),
title="My Custom Stable Diffusion img2img Interface",
description="This is a custom interface for Stable Diffusion img2img model with additional processing."
)
iface.launch()