Gabiburt commited on
Commit
028fa1b
1 Parent(s): dc010b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -12,7 +12,7 @@ class Dummy():
12
  pass
13
 
14
 
15
- pipeline_path = snapshot_download(repo_id='briaai/BRIA-2.4')
16
  sys.path.append(pipeline_path)
17
  from ella_xl_pipeline import EllaXLPipeline
18
 
@@ -37,7 +37,7 @@ def tocuda():
37
  pipe.pipe.unet.ella.to('cuda')
38
 
39
 
40
- # print("Optimizing BRIA-2.4 - this could take a while")
41
  # t=time.time()
42
  # pipe.unet = torch.compile(
43
  # pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
@@ -106,12 +106,11 @@ css = """
106
  """
107
  with gr.Blocks(css=css) as demo:
108
  with gr.Column(elem_id="col-container"):
109
- gr.Markdown("## BRIA 2.4")
110
  gr.HTML('''
111
  <p style="margin-bottom: 10px; font-size: 94%">
112
  This is a demo for
113
- <a href="https://huggingface.co/briaai/BRIA-2.4" target="_blank">BRIA 2.4 text-to-image </a>.
114
- BRIA 2.4 improve the generation of humans and illustrations compared to BRIA 2.2 while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
115
  </p>
116
  ''')
117
  with gr.Group():
@@ -122,7 +121,7 @@ with gr.Blocks(css=css) as demo:
122
  steps = gr.Textbox(label="Steps", value=50)
123
  negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
124
  submit_btn = gr.Button("Generate")
125
- result = gr.Image(label="BRIA-2.4 Result")
126
 
127
  # gr.Examples(
128
  # examples = [
 
12
  pass
13
 
14
 
15
+ pipeline_path = snapshot_download(repo_id='briaai/BRIA-2.3-T5')
16
  sys.path.append(pipeline_path)
17
  from ella_xl_pipeline import EllaXLPipeline
18
 
 
37
  pipe.pipe.unet.ella.to('cuda')
38
 
39
 
40
+ # print("Optimizing BRIA-2.3-T5 - this could take a while")
41
  # t=time.time()
42
  # pipe.unet = torch.compile(
43
  # pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
 
106
  """
107
  with gr.Blocks(css=css) as demo:
108
  with gr.Column(elem_id="col-container"):
109
+ gr.Markdown("## BRIA 2.3 T5")
110
  gr.HTML('''
111
  <p style="margin-bottom: 10px; font-size: 94%">
112
  This is a demo for
113
+ <a href="https://huggingface.co/briaai/BRIA-2.3-T5" target="_blank">BRIA 2.3 T5 text-to-image </a>.
 
114
  </p>
115
  ''')
116
  with gr.Group():
 
121
  steps = gr.Textbox(label="Steps", value=50)
122
  negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
123
  submit_btn = gr.Button("Generate")
124
+ result = gr.Image(label="BRIA-2.3-T5 Result")
125
 
126
  # gr.Examples(
127
  # examples = [