Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
|
|
| 14 |
|
| 15 |
|
| 16 |
def generate(
|
| 17 |
-
prompt, history, max_new_tokens=512, repetition_penalty=1.2, temperature=0.5, top_p=1, top_k=1
|
| 18 |
):
|
| 19 |
|
| 20 |
input_text = f"{prompt}, {history}"
|
|
@@ -23,13 +23,12 @@ def generate(
|
|
| 23 |
better_prompt = tokenizer.decode(outputs[0])
|
| 24 |
return better_prompt
|
| 25 |
|
| 26 |
-
|
| 27 |
additional_inputs=[
|
| 28 |
-
gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens"
|
| 29 |
-
gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty"
|
| 30 |
-
gr.Slider(value=0.5, minimum=0, maximum=1, step=0.05, interactive=True, label="Temperature"
|
| 31 |
-
gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, label="Top P"
|
| 32 |
-
gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K"
|
| 33 |
]
|
| 34 |
|
| 35 |
examples=[["Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.", None, None ]]
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
def generate(
|
| 17 |
+
prompt, history, max_new_tokens=512, repetition_penalty=1.2, temperature=0.5, top_p=1, top_k=1
|
| 18 |
):
|
| 19 |
|
| 20 |
input_text = f"{prompt}, {history}"
|
|
|
|
| 23 |
better_prompt = tokenizer.decode(outputs[0])
|
| 24 |
return better_prompt
|
| 25 |
|
|
|
|
| 26 |
additional_inputs=[
|
| 27 |
+
gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens"),
|
| 28 |
+
gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty"),
|
| 29 |
+
gr.Slider(value=0.5, minimum=0, maximum=1, step=0.05, interactive=True, label="Temperature"),
|
| 30 |
+
gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, label="Top P"),
|
| 31 |
+
gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K"),
|
| 32 |
]
|
| 33 |
|
| 34 |
examples=[["Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.", None, None ]]
|