thedevvikas commited on
Commit
57d0789
·
1 Parent(s): db0e3c3

fix: Dependency issue

Browse files
Files changed (4) hide show
  1. .python-version +1 -0
  2. app.py +5 -6
  3. pyproject.toml +10 -0
  4. uv.lock +0 -0
.python-version ADDED
@@ -0,0 +1 @@
 
 
1
+ 3.11
app.py CHANGED
@@ -9,21 +9,20 @@ from fastai.vision.all import PILImage
9
  from fastai.vision.all import load_learner
10
  from pathlib import Path
11
  import gradio as gr
12
- from nbdev.export import nb_export
13
 
14
- # %% dog-or-cat.ipynb 3
15
  # Function to check if it is a cat
16
  def is_cat(x): return x[0].isupper()
17
 
18
- # %% dog-or-cat.ipynb 5
19
  # Load the trained model
20
  learner = load_learner(fname="./dog_or_cat.pkl")
21
 
22
- # %% dog-or-cat.ipynb 6
23
  # Test the model
24
  learner.predict(image)
25
 
26
- # %% dog-or-cat.ipynb 7
27
  categories = ("Cat", "Dog")
28
 
29
  def classify_image(img):
@@ -31,7 +30,7 @@ def classify_image(img):
31
  print(pred, idx)
32
  return dict(zip(categories, map(float, probs)))
33
 
34
- # %% dog-or-cat.ipynb 9
35
  example_imgs_path = Path("../data/test_images/examples")
36
  img = gr.Image()
37
  label = gr.Label()
 
9
  from fastai.vision.all import load_learner
10
  from pathlib import Path
11
  import gradio as gr
 
12
 
13
+ # %% dog-or-cat.ipynb 4
14
  # Function to check if it is a cat
15
  def is_cat(x): return x[0].isupper()
16
 
17
+ # %% dog-or-cat.ipynb 6
18
  # Load the trained model
19
  learner = load_learner(fname="./dog_or_cat.pkl")
20
 
21
+ # %% dog-or-cat.ipynb 7
22
  # Test the model
23
  learner.predict(image)
24
 
25
+ # %% dog-or-cat.ipynb 8
26
  categories = ("Cat", "Dog")
27
 
28
  def classify_image(img):
 
30
  print(pred, idx)
31
  return dict(zip(categories, map(float, probs)))
32
 
33
+ # %% dog-or-cat.ipynb 10
34
  example_imgs_path = Path("../data/test_images/examples")
35
  img = gr.Image()
36
  label = gr.Label()
pyproject.toml ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ [project]
2
+ name = "minimal"
3
+ version = "0.1.0"
4
+ description = "Add your description here"
5
+ readme = "README.md"
6
+ requires-python = ">=3.11"
7
+ dependencies = [
8
+ "fastai>=2.7.19",
9
+ "gradio>=5.23.1",
10
+ ]
uv.lock ADDED
The diff for this file is too large to render. See raw diff