Yaxh64 commited on
Commit
e8d01a2
·
verified ·
1 Parent(s): ebd3e12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -2,12 +2,22 @@ import io
2
  import time
3
  from PIL import Image
4
  from fastapi import FastAPI, UploadFile, File
 
5
  from transformers import pipeline
6
  import uvicorn
7
 
8
  app = FastAPI()
9
 
10
- # Load model
 
 
 
 
 
 
 
 
 
11
  detector = pipeline(
12
  "image-classification",
13
  model="google/vit-base-patch16-224"
@@ -47,6 +57,7 @@ async def detect_video(file: UploadFile = File(...)):
47
 
48
  start = time.time()
49
 
 
50
  elapsed = int((time.time() - start) * 1000)
51
 
52
  return {
 
2
  import time
3
  from PIL import Image
4
  from fastapi import FastAPI, UploadFile, File
5
+ from fastapi.middleware.cors import CORSMiddleware
6
  from transformers import pipeline
7
  import uvicorn
8
 
9
  app = FastAPI()
10
 
11
+ # Enable CORS so frontend on Vercel can access backend
12
+ app.add_middleware(
13
+ CORSMiddleware,
14
+ allow_origins=["*"],
15
+ allow_credentials=True,
16
+ allow_methods=["*"],
17
+ allow_headers=["*"],
18
+ )
19
+
20
+ # Load model once at startup
21
  detector = pipeline(
22
  "image-classification",
23
  model="google/vit-base-patch16-224"
 
57
 
58
  start = time.time()
59
 
60
+ # placeholder video detection
61
  elapsed = int((time.time() - start) * 1000)
62
 
63
  return {