File size: 235 Bytes
cf44863
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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()