Update app.py

app.py

@@ -8,10 +8,6 @@ import PIL.Image
 import torch
 from diffusers import DiffusionPipeline, UNet2DConditionModel, LCMScheduler
 
-DESCRIPTION = 'This space is an API service meant to be used by frontend applications.'
-if not torch.cuda.is_available():
-    DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
-
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = int(os.getenv('MAX_IMAGE_SIZE', '1024'))
 SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
@@ -70,7 +66,13 @@ def generate(prompt: str,
                 output_type='pil').images[0]
 
 with gr.Blocks() as demo:
-    gr.Markdown(DESCRIPTION)
+    gr.HTML("""
+    <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
+        <div style="text-align: center; color: black;">
+            <p style="color: black;">This space is a REST API to programmatically generate MP4s using a LoRA.</p>
+            <p style="color: black;">Please see the <a href="https://hotshot.co" target="_blank">README.md</a> for more information.</p>
+        </div>
+    </div>""")
     secret_token = gr.Text(
         label='Secret Token',
         max_lines=1,