Hugging Face's logo

Spaces:
wiklif
/
animals_detection
Sleeping

App Files Community
wiklif commited on
Commit
65f60ce
·
1 Parent(s): fa79dd5

next update requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +7 -7
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import os
3
  from train_model import train
4
  from predict_model import predict_all
 
5
 
6
  def train_model():
7
  train()
 
2
  import os
3
  from train_model import train
4
  from predict_model import predict_all
5
+ os.environ['NUMPY_EXPERIMENTAL_ARRAY_FUNCTION'] = '0'
6
 
7
  def train_model():
8
  train()
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch
2
- torchvision
3
- gradio
4
- Pillow
5
- numpy
6
- matplotlib
7
- scikit-learn
 
1
+ torch==1.12.0
2
+ torchvision==0.13.0
3
+ gradio==3.23.0
4
+ Pillow==9.3.0
5
+ numpy==1.23.5
6
+ matplotlib==3.6.2
7
+ scikit-learn==1.1.3