Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import threading | |
| import os | |
| import torch | |
| os.environ["OMP_NUM_THREADS"] = str(os.cpu_count()) | |
| torch.set_num_threads(os.cpu_count()) | |
| model1 = gr.load("models/prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA") | |
| model2 = gr.load("models/Purz/face-projection") | |
| stop_event = threading.Event() | |
| def generate_images(text, selected_model): | |
| stop_event.clear() | |
| if selected_model == "Model 1 (Turbo Realism)": | |
| model = model1 | |
| elif selected_model == "Model 2 (Face Projection)": | |
| model = model2 | |
| else: | |
| return ["Invalid model selection."] * 3 | |
| results = [] | |
| for i in range(3): | |
| if stop_event.is_set(): | |
| return ["Image generation stopped by user."] * 3 | |
| modified_text = f"{text} variation {i+1}" | |
| model_output = model(modified_text) | |
| # If the returned object is a tuple, grab the first element: | |
| # e.g., model_output might look like: (pil_image, something_else) | |
| if isinstance(model_output, tuple): | |
| pil_image = model_output[0] | |
| # If the returned object is a dict with a key like 'image' or 'images', extract accordingly: | |
| elif isinstance(model_output, dict) and "images" in model_output: | |
| pil_image = model_output["images"][0] | |
| else: | |
| # Otherwise assume the model_output is directly the PIL image | |
| pil_image = model_output | |
| results.append(pil_image) | |
| return results | |
| def stop_generation(): | |
| """Stops the ongoing image generation by setting the stop_event flag.""" | |
| stop_event.set() | |
| return ["Generation stopped."] * 3 | |
| with gr.Blocks() as interface:#... | |
| gr.Markdown( | |
| "### ⚠ Sorry for the inconvenience. The Space is currently running on the CPU, which might affect performance. We appreciate your understanding." | |
| ) | |
| text_input = gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt...") | |
| model_selector = gr.Radio( | |
| ["Model 1 (Turbo Realism)", "Model 2 (Face Projection)"], | |
| label="Select Model", | |
| value="Model 1 (Turbo Realism)" | |
| ) | |
| with gr.Row(): | |
| generate_button = gr.Button("Generate 3 Images 🎨") | |
| stop_button = gr.Button("Stop Image Generation") | |
| with gr.Row(): | |
| output1 = gr.Image(label="Generated Image 1") | |
| output2 = gr.Image(label="Generated Image 2") | |
| output3 = gr.Image(label="Generated Image 3") | |
| generate_button.click(generate_images, inputs=[text_input, model_selector], outputs=[output1, output2, output3]) | |
| stop_button.click(stop_generation, inputs=[], outputs=[output1, output2, output3]) | |
| interface.launch() |