Spaces:
Sleeping
Sleeping
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
|