mirror of
https://github.com/ParisNeo/lollms.git
synced 2024-12-19 04:37:54 +00:00
fixed whisper install
This commit is contained in:
parent
3741be3f3a
commit
939d80ddd1
@ -7,7 +7,7 @@ from pathlib import Path
|
||||
from lollms.app import LollmsApplication
|
||||
from lollms.paths import LollmsPaths
|
||||
from lollms.config import TypedConfig, ConfigTemplate, BaseConfig
|
||||
from lollms.utilities import PackageManager
|
||||
from lollms.utilities import PackageManager, install_conda_package
|
||||
from lollms.stt import LollmsSTT
|
||||
from dataclasses import dataclass
|
||||
from PIL import Image, PngImagePlugin
|
||||
@ -19,6 +19,7 @@ from lollms.paths import LollmsPaths
|
||||
import subprocess
|
||||
|
||||
if not PackageManager.check_package_installed("whisper"):
|
||||
install_conda_package("ffmpeg")
|
||||
PackageManager.install_package("whisper")
|
||||
import whisper
|
||||
|
||||
|
@ -56,6 +56,26 @@ def install_conda_utility():
|
||||
# Create a new Conda environment with the specified Python version
|
||||
#run_command(Commands.CREATE, "-n", env_name, f"python={python_version}")
|
||||
|
||||
def install_conda_package(package_name):
|
||||
try:
|
||||
import platform
|
||||
if platform.system()=="Windows":
|
||||
conda_path = Path(sys.executable).parent.parent/"miniconda3"/"condabin"/"conda"
|
||||
else:
|
||||
conda_path = Path(sys.executable).parent.parent.parent/"miniconda3"/"bin"/"conda"
|
||||
ASCIIColors.red("Conda path:")
|
||||
ASCIIColors.yellow(conda_path)
|
||||
process = subprocess.Popen(f'{conda_path} install {package_name} -y', shell=True)
|
||||
|
||||
# Wait for the process to finish
|
||||
process.wait()
|
||||
#from conda.cli.python_api import run_command, Commands
|
||||
# Create a new Conda environment with the specified Python version
|
||||
#run_command(Commands.CREATE, "-n", env_name, f"python={python_version}")
|
||||
return True
|
||||
except Exception as ex:
|
||||
trace_exception(ex)
|
||||
return False
|
||||
|
||||
def create_conda_env(env_name, python_version):
|
||||
from lollms.security import sanitize_shell_code
|
||||
|
Loading…
Reference in New Issue
Block a user