rogerxavier commited on
Commit
26fc5e2
1 Parent(s): a7db268

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +18 -24
api.py CHANGED
@@ -8,41 +8,35 @@ import io
8
  import os
9
  from moviepyTest import test
10
 
11
- # app = FastAPI()
12
 
13
- # @app.get("/inference")
14
- # def inference():
15
- # return "<p>Hello, World!</p>"
16
 
17
- # @app.get("/infer_t5")
18
- # def t5(input):
19
 
20
- # return {"output": input}
21
 
22
- # @app.get("/moviepyTest")
23
- # def t5():
24
- # result = test()
25
 
26
- # return {"output": result}
27
 
28
 
29
 
30
- # @app.post("/getCoordinates")
31
- # async def getCoordinates(image: UploadFile = File(...)):
32
- # img = await image.read()
33
- # image = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
34
- # path_to_image = "save.jpg"
35
- # image.save(path_to_image)
36
- # return "获取图片保存成功"
37
 
38
 
39
 
40
 
41
- from flask import Flask, request, jsonify, send_file
42
- from io import BytesIO
43
 
44
- app = Flask(__name__)
45
 
46
- @app.route('/test', methods=['GET'])
47
- def test():
48
- return "123test"
 
8
  import os
9
  from moviepyTest import test
10
 
11
+ app = FastAPI()
12
 
13
+ @app.get("/inference")
14
+ def inference():
15
+ return "<p>Hello, World!</p>"
16
 
17
+ @app.get("/infer_t5")
18
+ def t5(input):
19
 
20
+ return {"output": input}
21
 
22
+ @app.get("/moviepyTest")
23
+ def t5():
24
+ result = test()
25
 
26
+ return {"output": result}
27
 
28
 
29
 
30
+ @app.post("/getCoordinates")
31
+ async def getCoordinates(image: UploadFile = File(...)):
32
+ img = await image.read()
33
+ image = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
34
+ path_to_image = "save.jpg"
35
+ image.save(path_to_image)
36
+ return "获取图片保存成功"
37
 
38
 
39
 
40
 
 
 
41
 
 
42