AP123 commited on
Commit
a95e4b1
·
verified ·
1 Parent(s): c8963b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  from PIL import Image
6
  import numpy as np
7
 
8
- # Load the models
9
  controlnet = ControlNetModel.from_pretrained(
10
  "briaai/BRIA-2.2-ControlNet-Recoloring",
11
  torch_dtype=torch.float16
@@ -58,8 +58,8 @@ def generate_image(input_image, prompt, controlnet_conditioning_scale):
58
 
59
  return images[0]
60
 
61
- # Gradio Interface
62
- description = "Anything to Anything. Transform anything to anything. Allow an adjuster for controlnet scale."
63
 
64
  with gr.Blocks() as demo:
65
  gr.Markdown("<h1><center>Image Transformation with Bria Recolor ControlNet</center></h1>")
 
5
  from PIL import Image
6
  import numpy as np
7
 
8
+ # Load models
9
  controlnet = ControlNetModel.from_pretrained(
10
  "briaai/BRIA-2.2-ControlNet-Recoloring",
11
  torch_dtype=torch.float16
 
58
 
59
  return images[0]
60
 
61
+ # Gradio
62
+ description = "Anything to Anything. Transform anything to anything with just a prompt!"
63
 
64
  with gr.Blocks() as demo:
65
  gr.Markdown("<h1><center>Image Transformation with Bria Recolor ControlNet</center></h1>")