to0ony commited on
Commit
56fc36d
·
1 Parent(s): a28bf85

minor adjustment

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.7, 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)
 
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)