Sankie005 commited on
Commit
b2de20b
·
verified ·
1 Parent(s): 610e5f2

Trying with 10

Browse files
Files changed (1) hide show
  1. app/main.py +33 -33
app/main.py CHANGED
@@ -1,33 +1,33 @@
1
- from fastapi import FastAPI, HTTPException,APIRouter,Request
2
- from pydantic import BaseModel
3
- from fastapi.responses import HTMLResponse
4
- from fastapi.staticfiles import StaticFiles
5
- from starlette.responses import FileResponse
6
- from fastapi.middleware.cors import CORSMiddleware
7
- import requests
8
- import json
9
- from transformers import pipeline
10
- app = FastAPI()
11
- origins = ["*"]
12
- app.add_middleware(
13
- CORSMiddleware,
14
- allow_origins=origins,
15
- )
16
- app.mount("/files/", StaticFiles(directory='../code'), name="index")
17
- @app.get("/")
18
- async def read_index():
19
- return FileResponse('app/html/index.html')
20
- @app.post("/Event_Query/")
21
- async def get_query(query,apikey):
22
- headers = {"X-API-Key": apikey}
23
- params = {"query": query}
24
- return requests.get(
25
- f"https://chat-api.you.com/smart?query={query}",
26
- params=params,
27
- headers=headers,
28
- ).json()
29
- @app.post("/Event_Summary/")
30
- async def get_summary(query):
31
- summarizer = pipeline("summarization")
32
- summary = summarizer(query, max_length=20, min_length=3, do_sample=False)
33
- return (summary[0]['summary_text'])
 
1
+ from fastapi import FastAPI, HTTPException,APIRouter,Request
2
+ from pydantic import BaseModel
3
+ from fastapi.responses import HTMLResponse
4
+ from fastapi.staticfiles import StaticFiles
5
+ from starlette.responses import FileResponse
6
+ from fastapi.middleware.cors import CORSMiddleware
7
+ import requests
8
+ import json
9
+ from transformers import pipeline
10
+ app = FastAPI()
11
+ origins = ["*"]
12
+ app.add_middleware(
13
+ CORSMiddleware,
14
+ allow_origins=origins,
15
+ )
16
+ app.mount("/files/", StaticFiles(directory='../code'), name="index")
17
+ @app.get("/")
18
+ async def read_index():
19
+ return FileResponse('app/html/index.html')
20
+ @app.post("/Event_Query/")
21
+ async def get_query(query,apikey):
22
+ headers = {"X-API-Key": apikey}
23
+ params = {"query": query}
24
+ return requests.get(
25
+ f"https://chat-api.you.com/smart?query={query}",
26
+ params=params,
27
+ headers=headers,
28
+ ).json()
29
+ @app.post("/Event_Summary/")
30
+ async def get_summary(query):
31
+ summarizer = pipeline("summarization")
32
+ summary = summarizer(query, max_length=10, min_length=10, do_sample=False)
33
+ return (summary[0]['summary_text'])