Jackss commited on
Commit
ac5a31f
·
1 Parent(s): d1659f7

Added app file

Browse files
Files changed (1) hide show
  1. main.py +8 -3
main.py CHANGED
@@ -2,6 +2,11 @@ from fastapi import FastAPI
2
 
3
  app = FastAPI()
4
 
5
- @app.get('/')
6
- def read_root():
7
- return {"Hello": "World!"}
 
 
 
 
 
 
2
 
3
  app = FastAPI()
4
 
5
+ # @app.get('/')
6
+ # def read_root():
7
+ # return {"Hello": "World!"}
8
+
9
+ @app.get('/similarity')
10
+ def similarity(input):
11
+ output = pipe_flan(input)
12
+ return {"output": output[0]["generated_text"]}