Sanjayraju30 commited on
Commit
fe19dec
Β·
verified Β·
1 Parent(s): 40afb3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -1,28 +1,28 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # βœ… Fast model
5
  generator = pipeline("text2text-generation", model="google/flan-t5-small")
6
 
7
  # 🧠 Prompt Template
8
  TEMPLATE = (
9
- "You are a polite, humble, and professional customer support agent. "
10
- "Respond to the following customer message:\n\n{input}\n\nReply:"
11
  )
12
 
13
  # πŸ” Generate Reply
14
  def generate_reply(user_input):
15
  prompt = TEMPLATE.format(input=user_input)
16
- response = generator(prompt, max_length=60, do_sample=False)[0]["generated_text"]
17
  return response.strip()
18
 
19
- # πŸŽ›οΈ Gradio 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="Polite Support Reply"),
24
- title="⚑ Ultra-Fast Auto-Reply Generator for Customer Support",
25
- description="Get polite, helpful replies in seconds using FLAN-T5-small. Built for speed + tone.",
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?"],
@@ -31,4 +31,4 @@ iface = gr.Interface(
31
  ]
32
  )
33
 
34
- iface.launch()
 
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?"],
 
31
  ]
32
  )
33
 
34
+ iface.launch()