Sanjayraju30 commited on
Commit
7096535
Β·
verified Β·
1 Parent(s): 1b73e6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,28 +1,28 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # πŸš€ Use lightweight, faster FLAN model
5
- generator = pipeline("text2text-generation", model="google/flan-t5-xs")
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
- # 🎯 Response Function
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="Fast & polite auto-replies to customer complaints using a tiny FLAN-T5 model. Perfect for CRM or helpdesk automation.",
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?"],