Emil25 commited on
Commit
66fd6ad
·
verified ·
1 Parent(s): 0ae8c85

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +103 -0
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()