vahidrezanezhad commited on
Commit
cffbee3
·
1 Parent(s): 5b0a672
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,5 +1,7 @@
1
  import gradio as gr
2
  import tensorflow as tf
 
 
3
 
4
  model = from_pretrained_keras("SBB/sbb_binarization")
5
 
 
1
  import gradio as gr
2
  import tensorflow as tf
3
+ from huggingface_hub import from_pretrained_keras
4
+
5
 
6
  model = from_pretrained_keras("SBB/sbb_binarization")
7
 
requirements.txt CHANGED
@@ -3,4 +3,4 @@ opencv-python
3
  tqdm
4
  pandas
5
  seaborn
6
- huggingface_hub["tensorflow"]
 
3
  tqdm
4
  pandas
5
  seaborn
6
+ huggingface_hub