Jofthomas HF staff commited on
Commit
1bfca77
1 Parent(s): abe9189

Figuring out how to blog other spaces from taping into this backend

Browse files
Files changed (1) hide show
  1. TextGen/router.py +2 -1
TextGen/router.py CHANGED
@@ -181,7 +181,8 @@ def story(prompt: Invoke):
181
  return inference_model(system_messsage=prompt.system_prompt,prompt=prompt.message)
182
 
183
  @app.post("/generate_level")
184
- def placement(input: Rooms):
 
185
  print(input)
186
  markdown_map=generate_map_markdown(input.rooms,input.room_of_interest,input.index_exit)
187
  story=generate_story(input.possible_entities)
 
181
  return inference_model(system_messsage=prompt.system_prompt,prompt=prompt.message)
182
 
183
  @app.post("/generate_level")
184
+ def placement(input: Rooms, request: Request):
185
+ client_ip = request.client.host
186
  print(input)
187
  markdown_map=generate_map_markdown(input.rooms,input.room_of_interest,input.index_exit)
188
  story=generate_story(input.possible_entities)