| 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) | |