xclasscode commited on
Commit
c23e33b
·
verified ·
1 Parent(s): ad38258

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +13 -0
main.py CHANGED
@@ -1,10 +1,23 @@
1
  from fastapi import Depends, FastAPI, HTTPException, status
2
  from sqlalchemy.orm import Session
 
3
  import schemas, models
4
  from transformers import pipeline
5
  import database
6
 
7
  app = FastAPI()
 
 
 
 
 
 
 
 
 
 
 
 
8
  models.database.Base.metadata.create_all(bind=database.engine)
9
  pipe = pipeline("text-classification", model="lxyuan/distilbert-base-multilingual-cased-sentiments-student")
10
  # Database session dependency
 
1
  from fastapi import Depends, FastAPI, HTTPException, status
2
  from sqlalchemy.orm import Session
3
+ from fastapi.middleware.cors import CORSMiddleware
4
  import schemas, models
5
  from transformers import pipeline
6
  import database
7
 
8
  app = FastAPI()
9
+ origins = [
10
+ "http://127.0.0.1",
11
+ ]
12
+
13
+ app.add_middleware(
14
+ CORSMiddleware,
15
+ allow_origins=http://localhost:5173,
16
+ allow_credentials=True,
17
+ allow_methods=["*"],
18
+ allow_headers=["*"],
19
+ )
20
+
21
  models.database.Base.metadata.create_all(bind=database.engine)
22
  pipe = pipeline("text-classification", model="lxyuan/distilbert-base-multilingual-cased-sentiments-student")
23
  # Database session dependency