Enhance image generation function by initializing random number generator for device-specific operations; update markdown instructions for clarity and improve HTML header formatting

app.py

@@ -507,6 +507,7 @@ def generate_image(prompt, cfg, tau, h_div_w, seed):
     torch.cuda.empty_cache()
 
     with torch.amp.autocast(device_type=device, dtype=autocast_dtype), torch.no_grad():
+        infinity.rng = torch.Generator(device=device)
         _, _, img_list = infinity.autoregressive_infer_cfg(
             vae=vae,
             scale_schedule=scale_schedule,
@@ -520,6 +521,7 @@ def generate_image(prompt, cfg, tau, h_div_w, seed):
             gt_leak=0, gt_ls_Bl=None, inference_mode=True,
             sampling_per_bits=args.sampling_per_bits,
         )
+        infinity.rng = torch.Generator(device="cpu")
     
     img = img_list[0]
     image = img.cpu().numpy()
@@ -531,18 +533,16 @@ def generate_image(prompt, cfg, tau, h_div_w, seed):
 
 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.
+1. Enter a detailed prompt with rich visual features or use the "Enhance Prompt" button to generate a more detailed description.
+2. Adjust the "CFG" and "Tau" sliders to control the strength and randomness of the output.
+3. Use the "Aspect Ratio" slider to set the aspect ratio of the generated image.
+4. Click the "Generate Image" button to create the image based on your 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>
+    <h1>Infinity Image Generator by <a href="https://github.com/FoundationVision/Infinity" target="_blank" rel="noopener noreferrer">FoundationVision</a></h1>
     <p style="font-size: 14px; color: #888;">This is not the official implementation from the main developers!</p>
 </div>"""
 with gr.Blocks() as demo: