soiz commited on
Commit
9d990ac
·
verified ·
1 Parent(s): 306e397

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
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