Update app.py

app.py

@@ -7,6 +7,7 @@ from screenshot import BG_COMP, BOX_COMP, GENERATION_VAR, PROMPT_VAR, main
 from pathlib import Path
 
 title = "BLOOM"
+
 description = """Gradio Demo for BLOOM. To use it, simply add your text, or click one of the examples to load them.
 Tips:
 - Do NOT talk to BLOOM as an entity, it's not a chatbot but a webpage/blog/article completion model.
@@ -17,6 +18,11 @@ Options:
 - greedy: accurate completions (may be more boring or have repetitions)
 """
 
+wip_description = """JAX / Flax Gradio Demo for BLOOM. The 176B BLOOM model running on a TPU v3-256 pod, with 2D model parallelism and custom mesh axes.
+
+Note: for this WIP demo only `sampling` is supported.
+"""
+
 API_URL =  os.getenv("API_URL")
 
 examples = [
@@ -90,8 +96,8 @@ gr.Interface(
     inference, 
     [
         gr.inputs.Textbox(label="Input"),
-        gr.inputs.Radio([32, 64], default=32, label="Tokens to generate"),
-        gr.inputs.Radio(["sampling", "greedy"], label="Sample or greedy", default="Sample"),
+        gr.inputs.Radio([64], default=64, label="Tokens to generate"),
+        gr.inputs.Radio(["sampling"], label="Sample or greedy", default="sampling"),
         gr.Checkbox(label="Just output raw text", value=True),
     ], 
     ["image", "text"],
@@ -99,5 +105,5 @@ gr.Interface(
     # article=article,
     cache_examples=False,
     title=title,
-    description=description
+    description=wip_description
 ).launch()