znchen commited on
Commit
eb81c16
1 Parent(s): 0946cca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -112,8 +112,8 @@ def rag_gen(
112
  run_num = update_run_num()
113
 
114
  # return image, rag_image, seed, f"<span style='font-size: 16px; font-weight: bold; color: red; display: block; text-align: center;'>Total inference runs: {run_num}</span>"
115
- return rag_image, seed, f"<span style='font-size: 16px; font-weight: bold; color: red; display: block; text-align: center;'>Total inference runs: {run_num}</span>"
116
-
117
 
118
  example_path = os.path.join(os.path.dirname(__file__), 'assets')
119
 
@@ -152,9 +152,9 @@ def load_description(fp):
152
  with gr.Blocks(css=css) as demo:
153
  gr.HTML(load_description("assets/title.md"))
154
 
155
- run_nums_box = gr.Markdown(
156
- value=f"<span style='font-size: 16px; font-weight: bold; color: red; display: block; text-align: center;'>Total inference runs: {run_num}</span>"
157
- )
158
 
159
  with gr.Row():
160
 
@@ -227,6 +227,11 @@ with gr.Blocks(css=css) as demo:
227
  Step 2. Press “Run” to get results
228
  </div>
229
  </div>
 
 
 
 
 
230
  """)
231
 
232
  # layout = gr.Image(label="Layout", show_label=True)
@@ -301,7 +306,8 @@ with gr.Blocks(css=css) as demo:
301
  randomize_seed,
302
  ],
303
  # outputs=[layout, result, seed, run_nums_box],
304
- outputs=[result, seed, run_nums_box],
 
305
  api_name="run",
306
  )
307
 
 
112
  run_num = update_run_num()
113
 
114
  # return image, rag_image, seed, f"<span style='font-size: 16px; font-weight: bold; color: red; display: block; text-align: center;'>Total inference runs: {run_num}</span>"
115
+ # return rag_image, seed, f"<span style='font-size: 16px; font-weight: bold; color: red; display: block; text-align: center;'>Total inference runs: {run_num}</span>"
116
+ return rag_image, seed
117
 
118
  example_path = os.path.join(os.path.dirname(__file__), 'assets')
119
 
 
152
  with gr.Blocks(css=css) as demo:
153
  gr.HTML(load_description("assets/title.md"))
154
 
155
+ # run_nums_box = gr.Markdown(
156
+ # value=f"<span style='font-size: 16px; font-weight: bold; color: red; display: block; text-align: center;'>Total inference runs: {run_num}</span>"
157
+ # )
158
 
159
  with gr.Row():
160
 
 
227
  Step 2. Press “Run” to get results
228
  </div>
229
  </div>
230
+ <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 10px;">
231
+ <div>
232
+ Errors may be displayed due to insufficient computing power
233
+ </div>
234
+ </div>
235
  """)
236
 
237
  # layout = gr.Image(label="Layout", show_label=True)
 
306
  randomize_seed,
307
  ],
308
  # outputs=[layout, result, seed, run_nums_box],
309
+ # outputs=[result, seed, run_nums_box],
310
+ outputs=[result, seed],
311
  api_name="run",
312
  )
313