Trang Dang commited on
Commit
eee3513
1 Parent(s): 83fe5b0

add requiremnts

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +3 -1
app.py CHANGED
@@ -67,7 +67,6 @@ def server(input: Inputs, output: Outputs, session: Session):
67
  # convert soft mask to hard mask
68
  single_patch_prob = single_patch_prob.cpu().numpy().squeeze()
69
  single_patch_prediction = (single_patch_prob > 0.5).astype(np.uint8)
70
-
71
  return img
72
  return None
73
 
 
67
  # convert soft mask to hard mask
68
  single_patch_prob = single_patch_prob.cpu().numpy().squeeze()
69
  single_patch_prediction = (single_patch_prob > 0.5).astype(np.uint8)
 
70
  return img
71
  return None
72
 
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
  shiny==0.9.0
2
  shinyswatch==0.6.0
3
  seaborn==0.12.2
4
- matplotlib==3.7.1
 
 
 
1
  shiny==0.9.0
2
  shinyswatch==0.6.0
3
  seaborn==0.12.2
4
+ matplotlib==3.7.1
5
+ transformers
6
+ torch