Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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.
|
|
|
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,
|