Spaces:
Sleeping
Sleeping
minor adjustment
Browse files
app.py
CHANGED
|
@@ -63,12 +63,12 @@ with gr.Blocks(title="🎬 Final Thesis Plot Generator") as demo:
|
|
| 63 |
|
| 64 |
model_choice = gr.Dropdown(
|
| 65 |
choices=["cmu-plots-model.pt", "cmu-plots-model-enchanced.pt"],
|
| 66 |
-
value="cmu-plots-model.pt",
|
| 67 |
label="Model"
|
| 68 |
)
|
| 69 |
prompt = gr.Textbox(label="Prompt", lines=5, placeholder="E.g. A young detective arrives in a coastal town...")
|
| 70 |
max_new_tokens = gr.Slider(32, 512, value=200, step=16, label="Max new tokens")
|
| 71 |
-
temperature = gr.Slider(0.1, 1.5, value=0.
|
| 72 |
top_k = gr.Slider(0, 100, value=50, step=5, label="Top-K (0 = off)")
|
| 73 |
btn = gr.Button("Generate")
|
| 74 |
output = gr.Textbox(label="Output", lines=15)
|
|
|
|
| 63 |
|
| 64 |
model_choice = gr.Dropdown(
|
| 65 |
choices=["cmu-plots-model.pt", "cmu-plots-model-enchanced.pt"],
|
| 66 |
+
value="cmu-plots-model-enchanced.pt",
|
| 67 |
label="Model"
|
| 68 |
)
|
| 69 |
prompt = gr.Textbox(label="Prompt", lines=5, placeholder="E.g. A young detective arrives in a coastal town...")
|
| 70 |
max_new_tokens = gr.Slider(32, 512, value=200, step=16, label="Max new tokens")
|
| 71 |
+
temperature = gr.Slider(0.1, 1.5, value=0.9, step=0.1, label="Temperature")
|
| 72 |
top_k = gr.Slider(0, 100, value=50, step=5, label="Top-K (0 = off)")
|
| 73 |
btn = gr.Button("Generate")
|
| 74 |
output = gr.Textbox(label="Output", lines=15)
|