Spaces:
Sleeping
Sleeping
Hemanth Kunta commited on
Commit ·
84b0d00
1
Parent(s): f2185c5
make reset and step accept empty POST bodies
Browse files- space_app.py +7 -3
space_app.py
CHANGED
|
@@ -6,7 +6,7 @@ import re
|
|
| 6 |
from typing import Any
|
| 7 |
|
| 8 |
import gradio as gr
|
| 9 |
-
from fastapi import FastAPI, HTTPException
|
| 10 |
|
| 11 |
from env.inprocess_backend import BACKEND
|
| 12 |
|
|
@@ -248,12 +248,16 @@ def _health() -> dict[str, str]:
|
|
| 248 |
|
| 249 |
|
| 250 |
@fastapi_app.post("/reset")
|
| 251 |
-
def _reset(payload: dict) -> dict:
|
|
|
|
|
|
|
|
|
|
| 252 |
return SESSION.reset(payload)
|
| 253 |
|
| 254 |
|
| 255 |
@fastapi_app.post("/step")
|
| 256 |
-
def _step(payload: dict) -> dict:
|
|
|
|
| 257 |
return SESSION.step(payload)
|
| 258 |
|
| 259 |
|
|
|
|
| 6 |
from typing import Any
|
| 7 |
|
| 8 |
import gradio as gr
|
| 9 |
+
from fastapi import Body, FastAPI, HTTPException
|
| 10 |
|
| 11 |
from env.inprocess_backend import BACKEND
|
| 12 |
|
|
|
|
| 248 |
|
| 249 |
|
| 250 |
@fastapi_app.post("/reset")
|
| 251 |
+
def _reset(payload: dict = Body(default_factory=dict)) -> dict:
|
| 252 |
+
payload = payload or {}
|
| 253 |
+
payload.setdefault("task_id", 1)
|
| 254 |
+
payload.setdefault("seed", 42)
|
| 255 |
return SESSION.reset(payload)
|
| 256 |
|
| 257 |
|
| 258 |
@fastapi_app.post("/step")
|
| 259 |
+
def _step(payload: dict = Body(default_factory=dict)) -> dict:
|
| 260 |
+
payload = payload or {}
|
| 261 |
return SESSION.step(payload)
|
| 262 |
|
| 263 |
|