Fabrice-TIERCELIN commited on
Commit
e14e150
1 Parent(s): 911fb02

HTML warning

Browse files
Files changed (1) hide show
  1. gradio_demo.py +7 -6
gradio_demo.py CHANGED
@@ -201,19 +201,20 @@ def submit_feedback(event_id, fb_score, fb_text):
201
 
202
  if torch.cuda.device_count() == 0:
203
  title_md = """
204
- # **SUPIR: Practicing Model Scaling for Photo-Realistic Image Restoration**
205
 
206
- ⚠️To use SUPIR, [Duplicate this space](https://huggingface.co/spaces/Fabrice-TIERCELIN/SUPIR?duplicate=true) and set a **GPU**.
207
 
208
- You can't use SUPIR directly here as this space runs on a CPU, which not enough for SUPIR. This is a template space. Please provide feedback if you have issues.
 
209
  """
210
  else:
211
  title_md = """
212
- # **SUPIR: Practicing Model Scaling for Photo-Realistic Image Restoration**
213
 
214
  ⚠️SUPIR is still a research project under tested and is not yet a stable commercial product.
215
 
216
- [[Paper](https://arxiv.org/abs/2401.13627)]   [[Project Page](http://supir.xpixel.group/)]   [[How to play](https://github.com/Fanghua-Yu/SUPIR/blob/master/assets/DemoGuide.png)]
217
  """
218
 
219
 
@@ -230,7 +231,7 @@ The service is a research preview intended for non-commercial use only, subject
230
  # Gradio interface
231
  with gr.Blocks(title='SUPIR') as interface:
232
  with gr.Row():
233
- gr.Markdown(title_md)
234
  with gr.Row():
235
  with gr.Column():
236
  with gr.Row(equal_height=True):
 
201
 
202
  if torch.cuda.device_count() == 0:
203
  title_md = """
204
+ <h1><center>SUPIR: Practicing Model Scaling for Photo-Realistic Image Restoration</center></h1>
205
 
206
+ <p style="color: red;"><big><big><big><b>⚠️To use SUPIR, <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/SUPIR?duplicate=true">Duplicate this space</a> and set a GPU with 60 GB VRAM.</b>
207
 
208
+ You can't use SUPIR directly here because this space runs on a CPU, which not enough for SUPIR. This is a template space. Please provide feedback if you have issues.
209
+ </big></big></big></p>
210
  """
211
  else:
212
  title_md = """
213
+ <h1><center>SUPIR: Practicing Model Scaling for Photo-Realistic Image Restoration</center></h1>
214
 
215
  ⚠️SUPIR is still a research project under tested and is not yet a stable commercial product.
216
 
217
+ <a href="https://arxiv.org/abs/2401.13627">Paper</a> &emsp; <a href="http://supir.xpixel.group/">Project Page</a> &emsp; <a href="https://github.com/Fanghua-Yu/SUPIR/blob/master/assets/DemoGuide.png">How to play</a>
218
  """
219
 
220
 
 
231
  # Gradio interface
232
  with gr.Blocks(title='SUPIR') as interface:
233
  with gr.Row():
234
+ gr.HTML(title_md)
235
  with gr.Row():
236
  with gr.Column():
237
  with gr.Row(equal_height=True):