indoflaven commited on
Commit
36443ae
·
1 Parent(s): bc67776

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -2,6 +2,7 @@ from fastapi import FastAPI
2
  from pydantic import BaseModel
3
  from typing import List
4
  from PIL import Image
 
5
 
6
  class Item(BaseModel):
7
  name: str
 
2
  from pydantic import BaseModel
3
  from typing import List
4
  from PIL import Image
5
+ from sentence_transformers import SentenceTransformer, util
6
 
7
  class Item(BaseModel):
8
  name: str