Spaces:
Runtime error
Runtime error
Commit
•
8413cb1
1
Parent(s):
3f37417
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ from screenshot import BG_COMP, BOX_COMP, GENERATION_VAR, PROMPT_VAR, main
|
|
7 |
from pathlib import Path
|
8 |
|
9 |
title = "BLOOM"
|
|
|
10 |
description = """Gradio Demo for BLOOM. To use it, simply add your text, or click one of the examples to load them.
|
11 |
Tips:
|
12 |
- 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:
|
|
17 |
- greedy: accurate completions (may be more boring or have repetitions)
|
18 |
"""
|
19 |
|
|
|
|
|
|
|
|
|
|
|
20 |
API_URL = os.getenv("API_URL")
|
21 |
|
22 |
examples = [
|
@@ -90,8 +96,8 @@ gr.Interface(
|
|
90 |
inference,
|
91 |
[
|
92 |
gr.inputs.Textbox(label="Input"),
|
93 |
-
gr.inputs.Radio([
|
94 |
-
gr.inputs.Radio(["sampling"
|
95 |
gr.Checkbox(label="Just output raw text", value=True),
|
96 |
],
|
97 |
["image", "text"],
|
@@ -99,5 +105,5 @@ gr.Interface(
|
|
99 |
# article=article,
|
100 |
cache_examples=False,
|
101 |
title=title,
|
102 |
-
description=
|
103 |
).launch()
|
|
|
7 |
from pathlib import Path
|
8 |
|
9 |
title = "BLOOM"
|
10 |
+
|
11 |
description = """Gradio Demo for BLOOM. To use it, simply add your text, or click one of the examples to load them.
|
12 |
Tips:
|
13 |
- Do NOT talk to BLOOM as an entity, it's not a chatbot but a webpage/blog/article completion model.
|
|
|
18 |
- greedy: accurate completions (may be more boring or have repetitions)
|
19 |
"""
|
20 |
|
21 |
+
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.
|
22 |
+
|
23 |
+
Note: for this WIP demo only `sampling` is supported.
|
24 |
+
"""
|
25 |
+
|
26 |
API_URL = os.getenv("API_URL")
|
27 |
|
28 |
examples = [
|
|
|
96 |
inference,
|
97 |
[
|
98 |
gr.inputs.Textbox(label="Input"),
|
99 |
+
gr.inputs.Radio([64], default=64, label="Tokens to generate"),
|
100 |
+
gr.inputs.Radio(["sampling"], label="Sample or greedy", default="sampling"),
|
101 |
gr.Checkbox(label="Just output raw text", value=True),
|
102 |
],
|
103 |
["image", "text"],
|
|
|
105 |
# article=article,
|
106 |
cache_examples=False,
|
107 |
title=title,
|
108 |
+
description=wip_description
|
109 |
).launch()
|