ElmahdiJaouali commited on
Commit
d83a6e5
·
1 Parent(s): ecdf515

Fix: Add static files mounting to serve logo.png

Browse files
Files changed (1) hide show
  1. api/main.py +4 -0
api/main.py CHANGED
@@ -7,6 +7,7 @@ Provides endpoints for image upload and prediction
7
  from fastapi import FastAPI, File, UploadFile, HTTPException
8
  from fastapi.middleware.cors import CORSMiddleware
9
  from fastapi.responses import HTMLResponse
 
10
  import numpy as np
11
  import joblib
12
  import json
@@ -27,6 +28,9 @@ app.add_middleware(
27
  allow_headers=["*"],
28
  )
29
 
 
 
 
30
  # Global variables for models and artifacts
31
  model = None
32
  scaler = None
 
7
  from fastapi import FastAPI, File, UploadFile, HTTPException
8
  from fastapi.middleware.cors import CORSMiddleware
9
  from fastapi.responses import HTMLResponse
10
+ from fastapi.staticfiles import StaticFiles
11
  import numpy as np
12
  import joblib
13
  import json
 
28
  allow_headers=["*"],
29
  )
30
 
31
+ # Mount static files
32
+ app.mount("/static", StaticFiles(directory="static"), name="static")
33
+
34
  # Global variables for models and artifacts
35
  model = None
36
  scaler = None