wehub-resource-sync
adf0d17497
publish / version_or_publish (push) Waiting to run
storybook-build / changes (push) Waiting to run
storybook-build / :storybook-build (push) Blocked by required conditions
Sync Gradio Skills to Hugging Face / sync-skills (push) Waiting to run
functional / changes (push) Waiting to run
functional / build-frontend (push) Blocked by required conditions
functional / functional-test-SSR=false (push) Blocked by required conditions
functional / functional-reload (push) Blocked by required conditions
functional / functional-test-SSR=true (push) Blocked by required conditions
hygiene / hygiene-test (push) Waiting to run
python / changes (push) Waiting to run
python / build (push) Blocked by required conditions
python / test-ubuntu-latest-flaky (push) Blocked by required conditions
python / test-ubuntu-latest-not-flaky (push) Blocked by required conditions
python / test-windows-latest-flaky (push) Blocked by required conditions
python / test-windows-latest-not-flaky (push) Blocked by required conditions
js / changes (push) Waiting to run
js / js-test (push) Blocked by required conditions
docs-build / changes (push) Waiting to run
docs-build / docs-build (push) Blocked by required conditions
docs-build / website-build (push) Blocked by required conditions
24 lines
628 B
Python
24 lines
628 B
Python
import torch
|
|
from diffusers import DiffusionPipeline # type: ignore
|
|
import gradio as gr
|
|
|
|
generator = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
|
|
# move to GPU if available
|
|
if torch.cuda.is_available():
|
|
generator = generator.to("cuda")
|
|
|
|
def generate(prompts):
|
|
images = generator(list(prompts)).images # type: ignore
|
|
return [images]
|
|
|
|
demo = gr.Interface(generate,
|
|
"textbox",
|
|
"image",
|
|
batch=True,
|
|
max_batch_size=4, # Set the batch size based on your CPU/GPU memory
|
|
api_name="predict"
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|