dominguezdaniel commited on
Commit
88dddb9
·
verified ·
1 Parent(s): 1c0ed94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -4,14 +4,21 @@ from transformers import pipeline
4
  generator = pipeline("text-generation", model="gpt2")
5
 
6
  def generate_description(business_name):
7
- prompt = f"Write a professional and SEO-friendly Google My Business description for a business named '{business_name}'. Highlight its quality, trust, and services."
 
 
 
 
8
  result = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
9
- return result
 
 
 
10
 
11
  gr.Interface(
12
  fn=generate_description,
13
- inputs=gr.Textbox(label="Business Name", placeholder="e.g. Webpy"),
14
  outputs="text",
15
- title="Google My Business Description Generator",
16
- description="Enter your business name and get a professional, AI-enhanced description for your GMB profile.",
17
  ).launch()
 
4
  generator = pipeline("text-generation", model="gpt2")
5
 
6
  def generate_description(business_name):
7
+ prompt = (
8
+ f"Create a Google My Business description for a business called '{business_name}'. "
9
+ "Make it sound commercial, professional, and persuasive. Highlight quality, customer trust, and top-rated services. "
10
+ "Use clear, natural language with strong marketing tone."
11
+ )
12
  result = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
13
+
14
+ # Remove leftover prompt or repetitions
15
+ cleaned = result.replace(prompt, "").strip()
16
+ return cleaned
17
 
18
  gr.Interface(
19
  fn=generate_description,
20
+ inputs=gr.Textbox(label="Business Name", placeholder="e.g. Pizza House"),
21
  outputs="text",
22
+ title="GMB Commercial Description Generator",
23
+ description="Generate a commercial, SEO-friendly description for your business profile. Just enter your business name.",
24
  ).launch()