suwonpabby commited on
Commit
6f3c591
·
1 Parent(s): d8e895c

First Landing

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import spaces
8
  import torch
9
  import uvicorn
10
  import time
11
- import numpy as np
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()