Spaces:
Running
Running
Commit
·
bddbfe8
1
Parent(s):
d7ec351
cors added
Browse files
main.py
CHANGED
@@ -9,7 +9,15 @@ from enum import Enum
|
|
9 |
from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration
|
10 |
import torch
|
11 |
import uvicorn
|
|
|
12 |
app = FastAPI(docs_url="/")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
14 |
|
15 |
|
|
|
9 |
from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration
|
10 |
import torch
|
11 |
import uvicorn
|
12 |
+
from fastapi.middleware.cors import CORSMiddleware
|
13 |
app = FastAPI(docs_url="/")
|
14 |
+
app.add_middleware(
|
15 |
+
CORSMiddleware,
|
16 |
+
allow_origins=["*"],
|
17 |
+
allow_methods=["*"],
|
18 |
+
allow_headers=["*"],
|
19 |
+
)
|
20 |
+
|
21 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
22 |
|
23 |
|