Spaces:
Build error
Build error
Trang Dang
commited on
Commit
·
eee3513
1
Parent(s):
83fe5b0
add requiremnts
Browse files- app.py +0 -1
- 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
|