joey1101 commited on
Commit
69c8028
Β·
verified Β·
1 Parent(s): 0a4b920

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -19,7 +19,7 @@ import sentencepiece # Tokenization dependency
19
  # Initial configuration (MUST be first)
20
  ##########################################
21
  st.set_page_config(
22
- page_title="πŸš€ Just Comment - AI Response Generator",
23
  page_icon="πŸ’¬",
24
  layout="centered",
25
  initial_sidebar_state="collapsed"
@@ -58,8 +58,8 @@ def load_models():
58
  ##########################################
59
  def render_interface():
60
  """Create user interface elements"""
61
- st.title("πŸš€ AI Customer Response Generator")
62
- st.caption("Analyzes feedback and generates tailored responses")
63
 
64
  return st.text_area(
65
  "πŸ“ Paste customer feedback here:",
@@ -128,7 +128,7 @@ def generate_text_response(user_input, models):
128
  inputs = models['textgen_tokenizer'](prompt, return_tensors="pt")
129
  outputs = models['textgen_model'].generate(
130
  inputs.input_ids,
131
- max_new_tokens=200,
132
  temperature=0.7,
133
  do_sample=True,
134
  top_p=0.9
 
19
  # Initial configuration (MUST be first)
20
  ##########################################
21
  st.set_page_config(
22
+ page_title="πŸš€ Just Comment",
23
  page_icon="πŸ’¬",
24
  layout="centered",
25
  initial_sidebar_state="collapsed"
 
58
  ##########################################
59
  def render_interface():
60
  """Create user interface elements"""
61
+ st.title("πŸš€ I'm listening to you, my friend~")
62
+ st.caption("Analyzes feedback and tell tailored responses")
63
 
64
  return st.text_area(
65
  "πŸ“ Paste customer feedback here:",
 
128
  inputs = models['textgen_tokenizer'](prompt, return_tensors="pt")
129
  outputs = models['textgen_model'].generate(
130
  inputs.input_ids,
131
+ max_new_tokens=100,
132
  temperature=0.7,
133
  do_sample=True,
134
  top_p=0.9