Spaces:
Running
Running
Update app.py
Browse files
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
|
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("π
|
62 |
-
st.caption("Analyzes feedback and
|
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=
|
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
|