Update app.py

app.py

@@ -26,20 +26,25 @@ def generate_images(prompt, num_inference_steps, guidance_scale, batch_size):
 # Define Gradio interface
 def get_cpu_info():
     cpu_name = psutil.cpu_freq().brand
-    memory_available = psutil.virtual_memory().available // 1024 // 1024
+    memory_available = psutil.virtual_memory().available // 1024 // 1024  # in MB
     return f"CPU: {cpu_name}, Memory: {memory_available} MB"
 
+cpu_info_text = gr.Textbox(label="CPU Information", value=get_cpu_info(), interactive=False)
+
 iface = gr.Interface(
     fn=generate_images,
     inputs=[
         gr.Textbox(label="Prompt"),
+        gr.Slider(label="Num Inference Steps", minimum=1, maximum=50, step=1, value=4),
         gr.Slider(label="Guidance Scale", minimum=0, maximum=20, step=0.1, value=0),
         gr.Slider(label="Batch Size", minimum=1, maximum=4, step=1, value=1),
     ],
-    outputs=gr.Gallery(label="Generated Images"),
+    outputs=[
+        gr.Gallery(label="Generated Images"),
+        cpu_info_text
+    ],
     title="SDXL Lightning 4-Step Inference (CPU)",
     description="Generate images with Stable Diffusion XL Lightning 4-Step model on CPU.",
-    extra_info=get_cpu_info,
 )
 
 iface.launch()