Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
from fastapi import FastAPI, Query
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
3 |
import torch
|
|
|
1 |
+
from fastapi.middleware.cors import CORSMiddleware
|
2 |
+
|
3 |
+
app = FastAPI()
|
4 |
+
|
5 |
+
# CORSの設定
|
6 |
+
app.add_middleware(
|
7 |
+
CORSMiddleware,
|
8 |
+
allow_origins=["*"], # すべてのオリジンを許可(必要に応じて制限を設定)
|
9 |
+
allow_credentials=True,
|
10 |
+
allow_methods=["*"], # すべてのHTTPメソッドを許可
|
11 |
+
allow_headers=["*"], # すべてのヘッダーを許可
|
12 |
+
)
|
13 |
+
|
14 |
+
|
15 |
from fastapi import FastAPI, Query
|
16 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
17 |
import torch
|