Spaces:
Sleeping
Sleeping
suwonpabby
commited on
Commit
·
6f3c591
1
Parent(s):
d8e895c
First Landing
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import spaces
|
|
8 |
import torch
|
9 |
import uvicorn
|
10 |
import time
|
11 |
-
|
12 |
|
13 |
# Initialize Logics
|
14 |
|
@@ -16,11 +16,13 @@ import numpy as np
|
|
16 |
os.system("pip install transformers")
|
17 |
os.system("pip install accelerate")
|
18 |
os.system("pip install -U FlagEmbedding")
|
|
|
19 |
|
20 |
# 2. Import Installed Packages
|
21 |
#fmt: off
|
22 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
23 |
from FlagEmbedding import BGEM3FlagModel
|
|
|
24 |
|
25 |
# 3. Initalize FastAPI App
|
26 |
app = FastAPI()
|
|
|
8 |
import torch
|
9 |
import uvicorn
|
10 |
import time
|
11 |
+
|
12 |
|
13 |
# Initialize Logics
|
14 |
|
|
|
16 |
os.system("pip install transformers")
|
17 |
os.system("pip install accelerate")
|
18 |
os.system("pip install -U FlagEmbedding")
|
19 |
+
os.system("pip install numpy")
|
20 |
|
21 |
# 2. Import Installed Packages
|
22 |
#fmt: off
|
23 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
24 |
from FlagEmbedding import BGEM3FlagModel
|
25 |
+
import numpy as np
|
26 |
|
27 |
# 3. Initalize FastAPI App
|
28 |
app = FastAPI()
|