Spaces:
Runtime error
Runtime error
Commit
·
e492d50
1
Parent(s):
f306f8a
use CPU, ruby default prompt
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ a 1.1B parameter model for code generation in Python, Java & JavaScript. The mod
|
|
10 |
with the <span style='color: #ff75b3;'><FILL-HERE></span> token.</span>"""
|
11 |
|
12 |
token = os.environ["HUB_TOKEN"]
|
13 |
-
device="
|
14 |
|
15 |
|
16 |
FIM_PREFIX = "<fim-prefix>"
|
@@ -97,7 +97,12 @@ with demo:
|
|
97 |
_, colum_2, _ = gr.Column(scale=1), gr.Column(scale=6), gr.Column(scale=1)
|
98 |
with colum_2:
|
99 |
gr.Markdown(value=description)
|
100 |
-
code = gr.Textbox(lines=5, label="Input code", value=
|
|
|
|
|
|
|
|
|
|
|
101 |
|
102 |
with gr.Accordion("Advanced settings", open=False):
|
103 |
max_new_tokens= gr.Slider(
|
@@ -126,5 +131,5 @@ with demo:
|
|
126 |
event = run.click(code_generation, [code, max_new_tokens, temperature, seed], output, api_name="predict")
|
127 |
gr.HTML(label="Contact", value="<img src='https://huggingface.co/datasets/bigcode/admin/resolve/main/bigcode_contact.png' alt='contact' style='display: block; margin: auto; max-width: 800px;'>")
|
128 |
|
129 |
-
#
|
130 |
demo.launch()
|
|
|
10 |
with the <span style='color: #ff75b3;'><FILL-HERE></span> token.</span>"""
|
11 |
|
12 |
token = os.environ["HUB_TOKEN"]
|
13 |
+
device="cpu"
|
14 |
|
15 |
|
16 |
FIM_PREFIX = "<fim-prefix>"
|
|
|
97 |
_, colum_2, _ = gr.Column(scale=1), gr.Column(scale=6), gr.Column(scale=1)
|
98 |
with colum_2:
|
99 |
gr.Markdown(value=description)
|
100 |
+
code = gr.Textbox(lines=5, label="Input code", value='''def fib(n)
|
101 |
+
if n <= 1
|
102 |
+
n
|
103 |
+
else
|
104 |
+
<FILL-HERE>
|
105 |
+
end''')
|
106 |
|
107 |
with gr.Accordion("Advanced settings", open=False):
|
108 |
max_new_tokens= gr.Slider(
|
|
|
131 |
event = run.click(code_generation, [code, max_new_tokens, temperature, seed], output, api_name="predict")
|
132 |
gr.HTML(label="Contact", value="<img src='https://huggingface.co/datasets/bigcode/admin/resolve/main/bigcode_contact.png' alt='contact' style='display: block; margin: auto; max-width: 800px;'>")
|
133 |
|
134 |
+
#demo.launch(share=True)
|
135 |
demo.launch()
|