Spaces:
Running
Running
fix 9
Browse files
app.py
CHANGED
@@ -31,8 +31,7 @@ def generate_text(
|
|
31 |
system_message,
|
32 |
max_tokens,
|
33 |
temperature,
|
34 |
-
top_p
|
35 |
-
do_sample
|
36 |
):
|
37 |
try:
|
38 |
print(f"Attempting to generate text for prompt: {prompt[:50]}...")
|
@@ -89,7 +88,7 @@ def respond(
|
|
89 |
|
90 |
"""
|
91 |
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
92 |
-
|
93 |
demo = gr.ChatInterface(
|
94 |
#respond,
|
95 |
generate_text,
|
@@ -106,6 +105,32 @@ demo = gr.ChatInterface(
|
|
106 |
),
|
107 |
],
|
108 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
109 |
|
110 |
|
111 |
if __name__ == "__main__":
|
|
|
31 |
system_message,
|
32 |
max_tokens,
|
33 |
temperature,
|
34 |
+
top_p
|
|
|
35 |
):
|
36 |
try:
|
37 |
print(f"Attempting to generate text for prompt: {prompt[:50]}...")
|
|
|
88 |
|
89 |
"""
|
90 |
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
91 |
+
|
92 |
demo = gr.ChatInterface(
|
93 |
#respond,
|
94 |
generate_text,
|
|
|
105 |
),
|
106 |
],
|
107 |
)
|
108 |
+
"""
|
109 |
+
|
110 |
+
with gr.Tab("Generate Email"):
|
111 |
+
Query = gr.Textbox(label="Query")
|
112 |
+
generate_button = gr.Button("Ask Query")
|
113 |
+
output = gr.Textbox(label="Generated Answer", lines=10)
|
114 |
+
|
115 |
+
generate_button.click(generate_text,
|
116 |
+
#inputs=[industry, recipient_role, company_details],
|
117 |
+
additional_inputs=[
|
118 |
+
Query,
|
119 |
+
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
120 |
+
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
121 |
+
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
122 |
+
gr.Slider(
|
123 |
+
minimum=0.1,
|
124 |
+
maximum=1.0,
|
125 |
+
value=0.95,
|
126 |
+
step=0.05,
|
127 |
+
label="Top-p (nucleus sampling)",
|
128 |
+
),
|
129 |
+
],
|
130 |
+
|
131 |
+
|
132 |
+
outputs=output)
|
133 |
+
|
134 |
|
135 |
|
136 |
if __name__ == "__main__":
|