Canopus-Flux-LoRA-Hoodies
The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.
Setting Up
import torch
from pipelines import DiffusionPipeline
base_model = "black-forest-labs/FLUX.1-dev"
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
lora_repo = "prithivMLmods/Canopus-Flux-LoRA-Hoodies"
trigger_word = "Cloth style" # Leave trigger_word blank if not used.
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
app.py
import gradio as gr
gr.load("prithivMLmods/Canopus-Flux-LoRA-Hoodies").launch()
pythonproject.py
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import gradio as gr
def image_generator(prompt):
pass
interface = gr.Interface(fn=image_generator, inputs="text", outputs="image")
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
app = gr.mount_gradio_app(app, interface, path="/")
App File Structure
/project-root/
β
βββ .gitattributes
βββ README.md
βββ app.py
βββ pythonproject.py
Trigger words
You should use Dress
to trigger the image generation.
You should use Cloth
to trigger the image generation.
You should use Cloth style
to trigger the image generation.
You should use Material Style
to trigger the image generation.
You should use Hoodie
to trigger the image generation.
Download model
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
- Downloads last month
- 0
Model tree for prithivMLmods/Canopus-Flux-LoRA-Hoodies
Base model
black-forest-labs/FLUX.1-dev