File size: 983 Bytes
4a26913 6cc8dc1 4a26913 6cc8dc1 4a26913 | 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 | import gradio as gr
import os, sys, subprocess
if not __package__:
from __init__ import Separator
else:
from .. import Separator
class Plugin(Separator):
def __init__(self):
self.name = "Тестовый плагин"
self.requirements = []
self.install_requirements(self.requirements)
def install_requirements(self, requirements: list):
if requirements:
cmd = [os.sys.executable, "-m", "pip", "install"]
for pkg in requirements:
cmd.append(pkg)
result = subprocess.run(cmd, text=True, capture_output=True)
def test(self):
print("Тест")
print(self.model_manager.get_mt())
def UI(self):
with gr.Column():
gr.Markdown("<h3><center>Пример рабочего плагина</center></h3>")
gr.Button("Показать все типы моделей", variant="primary").click(self.test)
|