import gradio as gr

model = gr.load("models/strangerzonehf/Flux-Midjourney-Mix2-LoRA")

def generate_image(prompt):
    return model.predict(prompt)

api = gr.Interface(fn=generate_image, inputs="text", outputs="image")

api.launch()