Files
viralfactory/src/engines/BaseEngine.py

20 lines
356 B
Python
Raw Normal View History

2024-02-13 14:15:27 +01:00
import gradio as gr
2024-02-15 17:53:02 +01:00
from abc import ABC, abstractmethod
from ..chore import GenerationContext
2024-02-13 14:15:27 +01:00
2024-02-14 17:49:51 +01:00
2024-02-13 14:15:27 +01:00
class BaseEngine(ABC):
2024-02-14 17:49:51 +01:00
num_options: int
2024-02-13 14:15:27 +01:00
name: str
description: str
def __init__(self):
2024-02-15 17:53:02 +01:00
self.ctx: GenerationContext # This is for type hinting only
2024-02-14 17:49:51 +01:00
pass
@classmethod
@abstractmethod
def get_options():
...