File size: 1,197 Bytes
a8b31c8 c3fe1fd a8b31c8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | import gradio as gr
from fooocus import FooocusProcessor # Adjust based on Fooocus setup
import os
print(os.listdir('.')) # To check the files in your Space directory
# Initialize Fooocus (update as needed for your model path or configs)
fooocus_processor = FooocusProcessor(model_path="path_to_model")
def process_with_fooocus(prompt, steps, guidance_scale):
"""
Generate an image using Fooocus based on user input.
"""
try:
result_image = fooocus_processor.generate(
prompt=prompt,
steps=steps,
guidance_scale=guidance_scale
)
return result_image
except Exception as e:
return f"An error occurred: {e}"
iface = gr.Interface(
fn=process_with_fooocus,
inputs=[
gr.Textbox(lines=2, placeholder="Enter your prompt here", label="Prompt"),
gr.Slider(1, 100, step=1, value=50, label="Steps"),
gr.Slider(1.0, 20.0, step=0.5, value=7.5, label="Guidance Scale"),
],
outputs=gr.Image(label="Generated Image"),
title="Fooocus Image Generator",
description="Generate high-quality images using Fooocus and Gradio."
)
if __name__ == "__main__":
iface.launch()
|