Enhance Gradio interface with detailed instructions and a custom HTML header for the Infinity Image Generator

app.py

@@ -502,9 +502,25 @@ def generate_image(prompt, cfg, tau, h_div_w, seed):
     return image
     
 
-# Set up Gradio interface
+markdown_description = """### Instructions:
+
+1. Enter a prompt in the text box below.
+2. Adjust the CFG (Classifier-Free Guidance) slider to control the strength of the prompt.
+3. Adjust the Tau (Temperature) slider to control the randomness of the output.
+4. Adjust the Aspect Ratio (Height/Width) slider to set the aspect ratio of the generated image.
+5. Click the Generate Image button to generate an image based on the prompt.
+
+Arxiv Paper:
+[Infinity: Scaling Bitwise AutoRegressive Modeling for High-Resolution Image Synthesis](https://arxiv.org/abs/2412.04431).
+"""
+html_header = """<div style="text-align: center; margin-bottom: 20px;">
+    <h1>Infinity Image Generator</h1>
+    <h2>by <a href="https://github.com/FoundationVision/Infinity" target="_blank" rel="noopener noreferrer">FoundationVision</a></h2>
+    <p style="font-size: 14px; color: #888;">This is not the official implementation from the main developers!</p>
+</div>"""
 with gr.Blocks() as demo:
-    gr.Markdown("<h1><center>Infinity Image Generator</center></h1>")
+    gr.HTML(html_header)
+    gr.Markdown(markdown_description)
     
     with gr.Row():
         with gr.Column():