codermert commited on
Commit
bc07329
1 Parent(s): 26f0892

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -29
README.md CHANGED
@@ -27,38 +27,11 @@ You should use `tugce` to trigger the image generation.
27
  ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
28
 
29
  ```py
30
- from fastapi import FastAPI, HTTPException
31
- from fastapi.responses import FileResponse
32
- import torch
33
  from diffusers import AutoPipelineForText2Image
34
- import io
35
-
36
- app = FastAPI()
37
-
38
- # Model ve LoRA'yı yükle
39
  pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
40
  pipeline.load_lora_weights('codermert/tugce2-lora', weight_name='flux_train_replicate.safetensors')
41
-
42
- @app.post("/generate_image")
43
- async def generate_image(prompt: str, width: int, height: int):
44
- try:
45
- image = pipeline(
46
- prompt,
47
- width=width,
48
- height=height
49
- ).images[0]
50
-
51
- img_byte_arr = io.BytesIO()
52
- image.save(img_byte_arr, format='PNG')
53
- img_byte_arr.seek(0)
54
-
55
- return FileResponse(img_byte_arr, media_type="image/png")
56
- except Exception as e:
57
- raise HTTPException(status_code=500, detail=str(e))
58
-
59
- if __name__ == "__main__":
60
- import uvicorn
61
- uvicorn.run(app, host="0.0.0.0", port=8000)
62
  ```
63
 
64
  For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)
 
27
  ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
28
 
29
  ```py
 
 
 
30
  from diffusers import AutoPipelineForText2Image
31
+ import torch
 
 
 
 
32
  pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
33
  pipeline.load_lora_weights('codermert/tugce2-lora', weight_name='flux_train_replicate.safetensors')
34
+ image = pipeline('your prompt').images[0]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
  ```
36
 
37
  For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)