gruhit-patel commited on
Commit
a00bdea
1 Parent(s): f33a809

Updated requirements

Browse files
Files changed (2) hide show
  1. backend.py +0 -1
  2. requirements.txt +3 -1
backend.py CHANGED
@@ -1,4 +1,3 @@
1
- import os
2
 
3
  from model import get_model
4
 
 
 
1
 
2
  from model import get_model
3
 
requirements.txt CHANGED
@@ -1,6 +1,8 @@
1
  streamlit
2
  fastapi
3
  uvicorn[standard]
4
- tensorflow==2.15.0
 
 
5
  pillow
6
  numpy
 
1
  streamlit
2
  fastapi
3
  uvicorn[standard]
4
+ torch==2.3.1
5
+ torchaudio==2.3.1
6
+ torchvision==0.18.1
7
  pillow
8
  numpy