Skip to content

Akilak11/moduled_project

Repository files navigation

Moduled Project

Moduled Project - это проект на основе Hugging Face Transformers, который включает несколько моделей для генерации текста. В проекте используются такие модели, как GPT-2 и T5. Весь функционал проекта разделен на отдельные модули для удобства использования и расширения.

Особенности проекта

  • Использование моделей GPT-2 и T5 для генерации текста
  • Разделение функционала на модули для легкого добавления новых моделей и функций
  • Возможность генерировать текст с различными параметрами, такими как temperature, num_beams, weights и другими
  • Простой пользовательский интерфейс для взаимодействия с моделями

Установка и настройка

  1. Клонируйте репозиторий:
git clone https://github.com/Akilak11/moduled_project.git
  1. Установите необходимые зависимости из файла requirements.txt:
pip install -r requirements.txt
  1. Загрузите предобученные модели GPT-2 и T5, если их нет локально, и разместите их в каталоге models/. Затем обновите пути к моделям в файлах gpt_2.py и t5.py.
  2. Запустите main.py для работы с проектом:
python main.py

Использование

При запуске main.py вы увидите пользовательский интерфейс, который предлагает выбрать модель и ввести текст для генерации. Выберите одну из доступных моделей, введите начальный текст, и программа сгенерирует продолжение текста на основе выбранной модели и заданных параметров.

Пример ввода:

Выберите модель (gpt-2, t5): gpt-2
Введите начальный текст: Привет, как дела?

Пример вывода:

Привет, как дела? У меня все отлично, спасибо за проявленный интерес! А как у тебя проходит день?

Разработка и дополнение проекта

Проект разделен на модули для легкого добавления новых моделей и функций. Вы можете расширить проект, добавляя новые модели или изменяя существующие, а также создавая новые функции или улучшая пользовательский интерфейс.

  • Для добавления новых моделей создайте новый файл в каталоге models/, например, my_new_model.py. Затем импортируйте вашу модель в user_interface.py и добавьте ее в список доступных моделей.
  • Если вам нужно добавить новые функции или оптимизировать существующие, вы можете изменять код в соответствующих модулях, а затем обновить user_interface.py для интеграции новых функций в пользовательский интерфейс.
  • Для улучшения пользовательского интерфейса вы можете изменить user_interface.py, добавлять новые опции, менять стиль отображения или добавлять новые инструкции для пользователей.

Поддержка и контакты

Если у вас возникли вопросы или предложения по улучшению проекта, пожалуйста, создайте issue в репозитории проекта на GitHub или свяжитесь с разработчиком по электронной почте.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages