Hemanth Kunta commited on
Commit
84b0d00
·
1 Parent(s): f2185c5

make reset and step accept empty POST bodies

Browse files
Files changed (1) hide show
  1. 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