AndriiPets commited on
Commit
8142ceb
·
1 Parent(s): 692fe6e

switched to fastapi

Browse files
Files changed (1) hide show
  1. main.py +13 -7
main.py CHANGED
@@ -25,16 +25,22 @@ def hello():
25
  return {"massage": "we are running!"}
26
 
27
 
28
- @app.get("/generate/")
29
- def generate(prompt: str, lang: str):
30
- # request_data = prompt.dict()
 
31
 
32
- # text = request_data["prompts"]
33
 
34
- if lang:
 
 
 
 
 
 
35
  generated_recepie = generation_function(
36
- prompt, lang=lang)
37
  else:
38
- generated_recepie = generation_function(prompt)
39
 
40
  return generated_recepie
 
25
  return {"massage": "we are running!"}
26
 
27
 
28
+ @app.get("/test")
29
+ def test(input: str):
30
+ output = input
31
+ return {"input": output}
32
 
 
33
 
34
+ @app.post("/generate/")
35
+ def generate(prompt: Prompt):
36
+ request_data = prompt.dict()
37
+
38
+ text = request_data["prompts"]
39
+
40
+ if request_data.get("lang"):
41
  generated_recepie = generation_function(
42
+ text, lang=request_data["lang"])
43
  else:
44
+ generated_recepie = generation_function(text)
45
 
46
  return generated_recepie