26 lines
608 B
Python
26 lines
608 B
Python
from .prediction import Prediction, ImageExpanderPredictionResponse
|
|
|
|
|
|
class ImageExpander(Prediction):
|
|
image: str
|
|
|
|
up_expansion_ratio: float
|
|
|
|
down_expansion_ratio: float
|
|
|
|
left_expansion_ratio: float
|
|
|
|
right_expansion_ratio: float
|
|
|
|
prompt: str
|
|
|
|
n: int
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._request_method = "POST"
|
|
self._request_path = "/v1/images/editing/expand"
|
|
self._query_prediction_info_method = "GET"
|
|
self._query_prediction_info_path = "/v1/images/editing/expand"
|
|
self._response_cls = ImageExpanderPredictionResponse
|