File size: 1,699 Bytes
bd91486 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | from setuptools import setup, find_packages
import subprocess
import yaml
import os
def setup_cogenbai():
"""Setup COGENBAI model"""
print("Setting up COGENBAI...")
# Create model
subprocess.run(["ollama", "create", "cogenbai:7.5b", "-f", "Modelfile"])
# Load configuration
with open("config.yaml", "r") as f:
config = yaml.safe_load(f)
print(f"\nCOGENBAI {config['model']['version']} setup complete!")
print("\nTest the model with:")
print("ollama run cogenbai:7.5b \"Write a hello world in Python\"")
if __name__ == "__main__":
setup_cogenbai()
setup(
name="cogenbai",
version="1.0.0",
packages=find_packages(),
install_requires=[
"torch>=1.9.0",
"transformers>=4.11.0",
"pyttsx3>=2.90",
"fastapi>=0.68.0",
"uvicorn>=0.15.0",
"click>=8.0.0",
"autopep8>=1.5.7",
"black>=21.5b2",
"yapf>=0.31.0",
"websockets>=10.0",
"python-socketio>=5.5.0"
],
entry_points={
'console_scripts': [
'cogenbai=cogenbai.cli.main:cli',
],
},
author="Shahrear Hossain Shawon",
author_email="contact@algoscienceacademy.com",
description="Advanced AI model for coding experts by Algo Science Academy",
long_description=open('README.md').read(),
long_description_content_type="text/markdown",
url="https://algoscienceacademy.com/cogenbai",
classifiers=[
"Development Status :: 5 - Production/Stable",
"Intended Audience :: Developers",
"License :: Other/Proprietary License",
"Programming Language :: Python :: 3.8",
],
python_requires=">=3.8",
)
|