chenguittiMaroua commited on
Commit
b649976
·
verified ·
1 Parent(s): 899e193

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +14 -5
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