Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -105,6 +105,18 @@ def get_qa_model():
|
|
105 |
#########################################################
|
106 |
|
107 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
108 |
# CORS Configuration
|
109 |
app.add_middleware(
|
110 |
CORSMiddleware,
|
@@ -132,10 +144,6 @@ SUPPORTED_FILE_TYPES = {
|
|
132 |
"png": "PNG Image"
|
133 |
}
|
134 |
|
135 |
-
|
136 |
-
|
137 |
-
|
138 |
-
|
139 |
MODEL_MAPPING = {
|
140 |
"fr": {
|
141 |
"qa": "illuin/camembert-base-fquad",
|
@@ -274,7 +282,6 @@ def preprocess_text(text: str) -> str:
|
|
274 |
text = re.sub(r'\s+', ' ', text).strip()
|
275 |
return text[:MAX_TEXT_LENGTH] if len(text) > MAX_TEXT_LENGTH else text
|
276 |
|
277 |
-
# API Endpoints
|
278 |
@app.post("/qa")
|
279 |
async def question_answering(
|
280 |
file: UploadFile = File(...),
|
@@ -329,6 +336,8 @@ async def question_answering(
|
|
329 |
|
330 |
# ... reste du code pour les questions normales ...
|
331 |
|
|
|
|
|
332 |
# Standard QA processing
|
333 |
result = qa_model(question=request.question, context=clean_text)
|
334 |
|
|
|
105 |
#########################################################
|
106 |
|
107 |
|
108 |
+
# Configure logging
|
109 |
+
logging.basicConfig(level=logging.INFO)
|
110 |
+
logger = logging.getLogger(__name__)
|
111 |
+
|
112 |
+
app = FastAPI(
|
113 |
+
title="AI Document Analysis API",
|
114 |
+
description="Advanced document processing with multilingual support",
|
115 |
+
version="2.0.0",
|
116 |
+
docs_url="/docs",
|
117 |
+
redoc_url="/redoc"
|
118 |
+
)
|
119 |
+
|
120 |
# CORS Configuration
|
121 |
app.add_middleware(
|
122 |
CORSMiddleware,
|
|
|
144 |
"png": "PNG Image"
|
145 |
}
|
146 |
|
|
|
|
|
|
|
|
|
147 |
MODEL_MAPPING = {
|
148 |
"fr": {
|
149 |
"qa": "illuin/camembert-base-fquad",
|
|
|
282 |
text = re.sub(r'\s+', ' ', text).strip()
|
283 |
return text[:MAX_TEXT_LENGTH] if len(text) > MAX_TEXT_LENGTH else text
|
284 |
|
|
|
285 |
@app.post("/qa")
|
286 |
async def question_answering(
|
287 |
file: UploadFile = File(...),
|
|
|
336 |
|
337 |
# ... reste du code pour les questions normales ...
|
338 |
|
339 |
+
# ... reste du code pour les questions normales ...
|
340 |
+
|
341 |
# Standard QA processing
|
342 |
result = qa_model(question=request.question, context=clean_text)
|
343 |
|