skylersterling commited on
Commit
5325787
·
verified ·
1 Parent(s): 23329aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,7 +13,7 @@ model.eval()
13
  model.to('cpu')
14
 
15
  # Define the function that generates text from a prompt
16
- def generate_text(prompt, temperature, top_p=1.0):
17
 
18
  print(prompt)
19
 
@@ -51,7 +51,8 @@ interface = gr.Interface(
51
  fn=generate_text,
52
  inputs=[
53
  gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
54
- gr.Slider(minimum=0.1, maximum=1.0, value=0.3, label="Temperature"),
 
55
  ],
56
  outputs=gr.Textbox(),
57
  live=False,
 
13
  model.to('cpu')
14
 
15
  # Define the function that generates text from a prompt
16
+ def generate_text(prompt, temperature, top_p):
17
 
18
  print(prompt)
19
 
 
51
  fn=generate_text,
52
  inputs=[
53
  gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
54
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.1, label="Temperature"),
55
+ gr.Slider(minimum=0.1, maximum=1.0, value=1.0, label="Top_P"),
56
  ],
57
  outputs=gr.Textbox(),
58
  live=False,