Spaces:
Sleeping
Sleeping
| from fastapi import FastAPI, HTTPException,APIRouter,Request | |
| from pydantic import BaseModel | |
| from fastapi.responses import HTMLResponse | |
| from fastapi.staticfiles import StaticFiles | |
| from starlette.responses import FileResponse | |
| from fastapi.middleware.cors import CORSMiddleware | |
| import requests | |
| import json | |
| from transformers import pipeline | |
| app = FastAPI() | |
| origins = ["*"] | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=origins, | |
| ) | |
| app.mount("/files/", StaticFiles(directory='../code'), name="index") | |
| async def read_index(): | |
| return FileResponse('app/html/index.html') | |
| async def get_query(query,apikey): | |
| headers = {"X-API-Key": apikey} | |
| params = {"query": query} | |
| return requests.get( | |
| f"https://chat-api.you.com/smart?query={query}", | |
| params=params, | |
| headers=headers, | |
| ).json() | |
| async def get_summary(query): | |
| summarizer = pipeline("summarization") | |
| summary = summarizer(query, max_length=10, min_length=10, do_sample=False) | |
| return (summary[0]['summary_text']) | |