Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,14 +35,14 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=devi
|
|
35 |
def post_processing(prompt, completion):
|
36 |
completion = "<span style='color: #ff75b3;'>" + completion + "</span>"
|
37 |
prompt = "<span style='color: #727cd6;'>" + prompt + "</span>"
|
38 |
-
code_html = f"<br><hr><br><pre style='
|
39 |
return GENERATION_TITLE + code_html
|
40 |
|
41 |
def post_processing_fim(prefix, middle, suffix):
|
42 |
prefix = "<span style='color: #727cd6;'>" + prefix + "</span>"
|
43 |
middle = "<span style='color: #ff75b3;'>" + middle + "</span>"
|
44 |
suffix = "<span style='color: #727cd6;'>" + suffix + "</span>"
|
45 |
-
code_html = f"<br><hr><br><pre style='
|
46 |
return GENERATION_TITLE + code_html
|
47 |
|
48 |
def fim_generation(prompt, max_new_tokens, temperature):
|
@@ -97,14 +97,14 @@ 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")
|
101 |
|
102 |
with gr.Accordion("Advanced settings", open=False):
|
103 |
max_new_tokens= gr.Slider(
|
104 |
minimum=8,
|
105 |
maximum=1024,
|
106 |
step=1,
|
107 |
-
value=
|
108 |
label="Number of tokens to generate",
|
109 |
)
|
110 |
temperature = gr.Slider(
|
|
|
35 |
def post_processing(prompt, completion):
|
36 |
completion = "<span style='color: #ff75b3;'>" + completion + "</span>"
|
37 |
prompt = "<span style='color: #727cd6;'>" + prompt + "</span>"
|
38 |
+
code_html = f"<br><hr><br><pre style='font-size: 12px'><code>{prompt}{completion}</code></pre><br><hr>"
|
39 |
return GENERATION_TITLE + code_html
|
40 |
|
41 |
def post_processing_fim(prefix, middle, suffix):
|
42 |
prefix = "<span style='color: #727cd6;'>" + prefix + "</span>"
|
43 |
middle = "<span style='color: #ff75b3;'>" + middle + "</span>"
|
44 |
suffix = "<span style='color: #727cd6;'>" + suffix + "</span>"
|
45 |
+
code_html = f"<br><hr><br><pre style='font-size: 12px'><code>{prefix}{middle}{suffix}</code></pre><br><hr>"
|
46 |
return GENERATION_TITLE + code_html
|
47 |
|
48 |
def fim_generation(prompt, max_new_tokens, temperature):
|
|
|
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 all_odd_elements(sequence):\n \"\"\"Returns every odd element of the sequence.\"\"\"")
|
101 |
|
102 |
with gr.Accordion("Advanced settings", open=False):
|
103 |
max_new_tokens= gr.Slider(
|
104 |
minimum=8,
|
105 |
maximum=1024,
|
106 |
step=1,
|
107 |
+
value=48,
|
108 |
label="Number of tokens to generate",
|
109 |
)
|
110 |
temperature = gr.Slider(
|