SANJAYV10 commited on
Commit
b65782e
1 Parent(s): 1e6b3cb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -0
requirements.txt CHANGED
@@ -2,5 +2,11 @@ fastapi
2
  pydantic
3
  keras
4
  h5py
 
 
 
 
 
 
5
  numpy
6
  uvicorn
 
2
  pydantic
3
  keras
4
  h5py
5
+ torch
6
+ albumentations
7
+ transformers
8
+ pytorch-lightning
9
+ torchmetrics
10
+ torchvision
11
  numpy
12
  uvicorn