Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
ADDED
@@ -0,0 +1,103 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import pandas as pd
|
3 |
+
|
4 |
+
|
5 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
6 |
+
|
7 |
+
|
8 |
+
# Настройка конфигурации страницы Streamlit
|
9 |
+
st.set_page_config(
|
10 |
+
page_title="Generate reviews",
|
11 |
+
initial_sidebar_state="expanded"
|
12 |
+
)
|
13 |
+
|
14 |
+
# Заголовок приложения
|
15 |
+
st.title("Генератор отзывов на основе ИИ")
|
16 |
+
st.write("Создайте уникальные текстовые отзывы о различных местах на основе категорий, рейтинга и ключевых слов.")
|
17 |
+
|
18 |
+
|
19 |
+
# Загрузка модели и токенизатора
|
20 |
+
# @st.cache_data()
|
21 |
+
@st.cache_resource
|
22 |
+
def get_model():
|
23 |
+
# Загрузка модели
|
24 |
+
model = AutoModelForCausalLM.from_pretrained('model')
|
25 |
+
# Загрузка токенизатора
|
26 |
+
tokenizer = AutoTokenizer.from_pretrained('model')
|
27 |
+
return model, tokenizer
|
28 |
+
|
29 |
+
|
30 |
+
# Генерация отзыва
|
31 |
+
def gen_review(input_text):
|
32 |
+
model, tokenizer = get_model()
|
33 |
+
input_ids = tokenizer.encode(input_text, return_tensors='pt')
|
34 |
+
output = model.generate(
|
35 |
+
input_ids,
|
36 |
+
max_length=200,
|
37 |
+
num_return_sequences=1,
|
38 |
+
no_repeat_ngram_size=2,
|
39 |
+
do_sample=True,
|
40 |
+
top_p=0.95,
|
41 |
+
top_k=60,
|
42 |
+
temperature=0.9,
|
43 |
+
eos_token_id=tokenizer.eos_token_id,
|
44 |
+
)
|
45 |
+
return tokenizer.decode(output[0], skip_special_tokens=True)
|
46 |
+
|
47 |
+
|
48 |
+
def capitalize_and_punctuate(text):
|
49 |
+
# Извлекаем часть текста после последнего двоеточия
|
50 |
+
text = text.split(":")[-1].strip()
|
51 |
+
|
52 |
+
# Разделяем текст на предложения по общим знакам препинания
|
53 |
+
sentences = []
|
54 |
+
current_sentence = []
|
55 |
+
for char in text:
|
56 |
+
current_sentence.append(char)
|
57 |
+
# Если встречаем знак конца предложения, добавляем его в список предложений
|
58 |
+
if char in '.!?':
|
59 |
+
sentences.append(''.join(current_sentence).strip())
|
60 |
+
current_sentence = []
|
61 |
+
|
62 |
+
# Если остался текст, добавляем его как последнее предложение
|
63 |
+
if current_sentence:
|
64 |
+
sentences.append(''.join(current_sentence).strip())
|
65 |
+
|
66 |
+
# Обрабатываем каждое предложение, чтобы сделать первую букву заглавной
|
67 |
+
corrected_sentences = []
|
68 |
+
for sentence in sentences:
|
69 |
+
if sentence:
|
70 |
+
# Делаем первую букву заглавной и добавляем точку в конце, если её нет
|
71 |
+
corrected_sentence = sentence[0].upper() + sentence[1:]
|
72 |
+
if not corrected_sentence.endswith('.'):
|
73 |
+
corrected_sentence += '.'
|
74 |
+
corrected_sentences.append(corrected_sentence)
|
75 |
+
|
76 |
+
# Объединяем все исправленные предложения в финальный текст
|
77 |
+
final_text = ' '.join(corrected_sentences)
|
78 |
+
return final_text
|
79 |
+
|
80 |
+
|
81 |
+
# Главная функция
|
82 |
+
def main():
|
83 |
+
|
84 |
+
if 'btn_predict' not in st.session_state:
|
85 |
+
st.session_state['btn_predict'] = False
|
86 |
+
|
87 |
+
category = st.text_input("Категория:", value="Кондитерская")
|
88 |
+
rating = st.slider("Рейтинг", 1, 5, 1)
|
89 |
+
key_words = st.text_input("Ключевые слова", value="десерт, торт, цена")
|
90 |
+
|
91 |
+
# Ввод новых параметров
|
92 |
+
input_text = f"Категория: {category}; Рейтинг: {rating}; Ключевые слова: {key_words} -> Отзыв:"
|
93 |
+
|
94 |
+
if st.button('Generate'):
|
95 |
+
with st.spinner('Генерация отзыва...'):
|
96 |
+
generated_text = gen_review(input_text)
|
97 |
+
generated_text = capitalize_and_punctuate(generated_text)
|
98 |
+
st.success("Готово!")
|
99 |
+
st.text(generated_text)
|
100 |
+
|
101 |
+
|
102 |
+
if __name__ == "__main__":
|
103 |
+
main()
|