Update README.md

README.md

@@ -6,22 +6,68 @@ language:
 pipeline_tag: text-generation
 library_name: transformers
 ---
-Стрела - большая языковая модель, разработанная с целья достижения большой скорости и качества ответов на слабых устройствах.
+Стрела — это мощная языковая модель, созданная для обеспечения высокой скорости работы и качества ответов на слабых устройствах.
 ## Описание от стрелы
 Я представляю собой компьютерную программу, которая была разработана для обработки и анализа естественного языка.
 Я обладаю способностью понимать, анализировать и обрабатывать естественный язык, что позволяет мне общаться с людьми через различные каналы коммуникации. Моя основная цель - помочь людям в решении задач и предоставить информацию на основе запроса.
 Я могу использоваться для различных целей: от автоматической генерации текста, перевода с одного языка на другой или даже создания собственных стихов и песен.
-## Использование модели
+## Использование модели для чата
 ```py
-from transformers import pipeline
+# Импорт библиотек
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-pipe = pipeline("text-generation", model="gai-labs/strela")
-chat = """### System:
-You are an AI assistant who gives a helpfull response to whatever human ask of you.
+# Загрузка модели
+tokenizer = AutoTokenizer.from_pretrained("gai-labs/strela")
+model = AutoModelForCausalLM.from_pretrained("gai-labs/strela")
+
+# Промпт системы
+system_prompt = "You are an AI assistant who gives a helpfull response to whatever human ask of you."
+
+# Ваш промпт
+prompt = "Привет!"
+
+# Шаблон чата
+chat = f"""### System:
+{system_prompt}
 ### Human:
-Привет!
+{prompt}
 ### Assistant:
 """
-outputs = pipe(chat, num_return_sequences=1, return_full_text=False)
-print(outputs)
+
+# Генерация
+model_inputs = tokenizer([chat], return_tensors="pt")
+generated_ids = model.generate(**model_inputs, max_new_tokens=64) # Настройте максимальное количество токенов для генерации
+output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+
+# Очистка выводов от шаблона чата
+output = output.replace(chat, "")
+
+# Вывод результатов генерации
+print(output)
+```
+```
+Привет! Чем могу помочь?
+```
+## Использование модели для генерации текста
+```py
+# Импорт библиотек
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# Загрузка модели
+tokenizer = AutoTokenizer.from_pretrained("gai-labs/strela")
+model = AutoModelForCausalLM.from_pretrained("gai-labs/strela")
+
+# Промпт
+prompt = "ИИ - "
+
+# Генерация
+model_inputs = tokenizer([prompt], return_tensors="pt")
+generated_ids = model.generate(**model_inputs, max_new_tokens=64) # Настройте максимальное количество токенов для генерации
+output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+
+# Вывод результатов генерации
+print(output)
+```
+```
+ИИ - это область компьютерных наук и технологий, которая занимается созданием машин, способных "понимать" человека или выполнять задачи с аналогичной логикой, как у людей.
 ```