Jofthomas HF staff commited on
Commit
90a0040
1 Parent(s): bc7f57e

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -168,7 +168,7 @@ def story(prompt: Invoke):
168
  return inference_model(system_messsage=prompt.system_prompt,prompt=prompt.message)
169
 
170
  @app.post("/generate_level")
171
- def placement(input: Room):
172
  print(input)
173
  answer={
174
  "key":"value"
 
168
  return inference_model(system_messsage=prompt.system_prompt,prompt=prompt.message)
169
 
170
  @app.post("/generate_level")
171
+ def placement(input: Rooms):
172
  print(input)
173
  answer={
174
  "key":"value"