Rectifier / App /Generate /utils /RenderVideo.py
Mbonea's picture
generation update
d65b1bc
raw
history blame
807 Bytes
import aiohttp, json
class RenderVideo:
def __init__(self, url="https://yakova-rectifier.hf.space/create-video"):
self.url = url
@staticmethod
def _parse_response(response: str) -> dict:
return json.loads(response)
async def post_request(self, data: dict) -> dict:
headers = {"Accept": "application/json", "Content-Type": "application/json"}
async with aiohttp.ClientSession() as session:
async with session.post(
self.url, data=json.dumps(data), headers=headers
) as resp:
response = await resp.text()
result = self._parse_response(response)
return result
async def render_video(self, data) -> dict:
result = await self.post_request(data)
return result