Update app.py
Browse files
app.py
CHANGED
@@ -46,18 +46,6 @@ def generate_response(prompt, model, tokenizer, max_length=100, num_beams=5, tem
|
|
46 |
# Create the Gradio chat interface
|
47 |
demo = gr.ChatInterface(
|
48 |
fn=generate_response,
|
49 |
-
additional_inputs=[
|
50 |
-
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
51 |
-
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
52 |
-
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
53 |
-
gr.Slider(
|
54 |
-
minimum=0.1,
|
55 |
-
maximum=1.0,
|
56 |
-
value=0.95,
|
57 |
-
step=0.05,
|
58 |
-
label="Top-p (nucleus sampling)",
|
59 |
-
),
|
60 |
-
],
|
61 |
title="Chatbot",
|
62 |
description="Ask anything to the chatbot."
|
63 |
)
|
|
|
46 |
# Create the Gradio chat interface
|
47 |
demo = gr.ChatInterface(
|
48 |
fn=generate_response,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
49 |
title="Chatbot",
|
50 |
description="Ask anything to the chatbot."
|
51 |
)
|