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?"],
|