Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,14 +29,14 @@ def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_len
|
|
29 |
iface = gr.Interface(
|
30 |
fn=generate_text,
|
31 |
inputs=[
|
32 |
-
gr.
|
33 |
-
gr.
|
34 |
-
gr.
|
35 |
-
gr.
|
36 |
-
gr.
|
37 |
-
gr.
|
38 |
],
|
39 |
-
outputs=gr.
|
40 |
title="Text Completion Model",
|
41 |
description="Try out the Mistral-7B model for free! Note this is the pretrained model and is not fine-tuned for instruction."
|
42 |
)
|
|
|
29 |
iface = gr.Interface(
|
30 |
fn=generate_text,
|
31 |
inputs=[
|
32 |
+
gr.Textbox(label="Prompt", lines=2, placeholder="Type a prompt..."),
|
33 |
+
gr.Slider(minimum=0.1, maximum=2.0, step=0.01, default=0.8, label="Temperature"),
|
34 |
+
gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="Top p"),
|
35 |
+
gr.Slider(minimum=0, maximum=100, step=1, default=40, label="Top k"),
|
36 |
+
gr.Slider(minimum=1.0, maximum=2.0, step=0.01, default=1.10, label="Repetition Penalty"),
|
37 |
+
gr.Slider(minimum=5, maximum=4096, step=5, default=1024, label="Max Length")
|
38 |
],
|
39 |
+
outputs=gr.Textbox(label="Generated Text"),
|
40 |
title="Text Completion Model",
|
41 |
description="Try out the Mistral-7B model for free! Note this is the pretrained model and is not fine-tuned for instruction."
|
42 |
)
|