Update app.py

app.py

@@ -1,11 +1,33 @@
-import json      
-import numpy as np      
-import tensorflow as tf      
-from tensorflow.keras import layers      
-import re    
-import requests    
-import math    
+import requests
+import numpy as np
+import tensorflow as tf
+from tensorflow.keras import layers  
+import asyncio
+from fastapi import FastAPI, Request
+from fastapi.responses import StreamingResponse, PlainTextResponse
 import sentencepiece as spm
+import re
+import math
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
+
+app = FastAPI()    
+
+
+from fastapi.middleware.cors import CORSMiddleware
+
+origins = [
+    "https://insect5386.github.io",
+    "https://insect5386.github.io/insect5386"
+]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 # SentencePiece 로드 (토크나이저랑 특수 토큰 ID도 동일하게 세팅)
 sp = spm.SentencePieceProcessor()