Spaces:
Sleeping
Sleeping
| from fastapi import APIRouter, UploadFile, File | |
| import openai | |
| import io | |
| import os | |
| from pypdf import PdfReader | |
| router = APIRouter() | |
| openai.api_key = os.getenv("OPENAI_API_KEY") | |
| async def embed_doc(file: UploadFile = File(...)): | |
| #for now just truncate based on length of words | |
| content = await file.read() | |
| return openai.Embedding.create(input = content.decode("utf-8"), model = "text-embedding-ada-002") |