Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,28 +1,28 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
#
|
5 |
-
generator = pipeline("text2text-generation", model="google/flan-t5-
|
6 |
|
7 |
-
#
|
8 |
TEMPLATE = (
|
9 |
"You are a polite and professional customer support agent. "
|
10 |
"Please respond to the customer's message:\n\n{input}"
|
11 |
)
|
12 |
|
13 |
-
#
|
14 |
def generate_reply(user_input):
|
15 |
prompt = TEMPLATE.format(input=user_input)
|
16 |
response = generator(prompt, max_length=80, do_sample=False)[0]["generated_text"]
|
17 |
return response.strip()
|
18 |
|
19 |
-
#
|
20 |
iface = gr.Interface(
|
21 |
fn=generate_reply,
|
22 |
inputs=gr.Textbox(lines=6, label="Customer Message", placeholder="Enter complaint or question..."),
|
23 |
outputs=gr.Textbox(label="Support Reply"),
|
24 |
title="β‘ Fast Auto-Reply Generator for Customer Support",
|
25 |
-
description="
|
26 |
examples=[
|
27 |
["I still haven't received my order and it's been 10 days."],
|
28 |
["Why was I charged twice for my subscription?"],
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
# β
Fast + available model
|
5 |
+
generator = pipeline("text2text-generation", model="google/flan-t5-small")
|
6 |
|
7 |
+
# π§ Prompt Template
|
8 |
TEMPLATE = (
|
9 |
"You are a polite and professional customer support agent. "
|
10 |
"Please respond to the customer's message:\n\n{input}"
|
11 |
)
|
12 |
|
13 |
+
# π Generate Reply
|
14 |
def generate_reply(user_input):
|
15 |
prompt = TEMPLATE.format(input=user_input)
|
16 |
response = generator(prompt, max_length=80, do_sample=False)[0]["generated_text"]
|
17 |
return response.strip()
|
18 |
|
19 |
+
# ποΈ Interface
|
20 |
iface = gr.Interface(
|
21 |
fn=generate_reply,
|
22 |
inputs=gr.Textbox(lines=6, label="Customer Message", placeholder="Enter complaint or question..."),
|
23 |
outputs=gr.Textbox(label="Support Reply"),
|
24 |
title="β‘ Fast Auto-Reply Generator for Customer Support",
|
25 |
+
description="Generate fast, polite, and professional replies to customer messages using Google's FLAN-T5.",
|
26 |
examples=[
|
27 |
["I still haven't received my order and it's been 10 days."],
|
28 |
["Why was I charged twice for my subscription?"],
|