Commit
•
e189f90
1
Parent(s):
65763b8
Uploaded the model script
Browse files- backend.py +1 -1
- model_script.pt +3 -0
backend.py
CHANGED
|
@@ -53,7 +53,7 @@ def get_prediction(image, model):
|
|
| 53 |
|
| 54 |
############## Backend #############
|
| 55 |
app = FastAPI()
|
| 56 |
-
model =
|
| 57 |
|
| 58 |
@app.get("/")
|
| 59 |
def foo():
|
|
|
|
| 53 |
|
| 54 |
############## Backend #############
|
| 55 |
app = FastAPI()
|
| 56 |
+
model = T.jit.load('model_script.pt')
|
| 57 |
|
| 58 |
@app.get("/")
|
| 59 |
def foo():
|
model_script.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a365e3b7032d30a4cafa616b1eddb4a95c22f27f1974fa02065b73ae35290729
|
| 3 |
+
size 257050510
|