Fabrice-TIERCELIN
commited on
Commit
•
e14e150
1
Parent(s):
911fb02
HTML warning
Browse files- 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 |
-
|
205 |
|
206 |
-
|
207 |
|
208 |
-
You can't use SUPIR directly here
|
|
|
209 |
"""
|
210 |
else:
|
211 |
title_md = """
|
212 |
-
|
213 |
|
214 |
⚠️SUPIR is still a research project under tested and is not yet a stable commercial product.
|
215 |
|
216 |
-
|
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.
|
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>   <a href="http://supir.xpixel.group/">Project Page</a>   <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):
|