Update TextGen/router.py
Browse files- 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:
|
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"
|