Hugging Face's logo

Spaces:
lf-netizen
/
lego_classification
Build error

App Files Community
LF-netizen commited on
Commit
fdb5b73
·
1 Parent(s): 118e0d5

add requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +1 -2
  2. requirements.txt +2 -0
app.py CHANGED
@@ -1,4 +1,3 @@
1
- import timm
2
  from fastai.vision.all import *
3
  import gradio as gr
4
  import os
@@ -18,7 +17,7 @@ I was especially intrested in how the model will do on MOCS a.k.a. community cre
18
  '''
19
 
20
  themes = sorted(('City', 'Technic', 'Star-Wars', 'Creator', 'Ninjago', 'Architecture', 'Duplo', 'Friends', 'DC-Comics-Super-Heroes'))
21
- learn_color = load_learner('models/lego_convnext_small_4ep_sets05-19.pkl')
22
  learn_gray = load_learner('models/lego_convnext_small_4ep_grayscale.pkl')
23
 
24
  def classify(img, is_color):
 
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
  import os
 
17
  '''
18
 
19
  themes = sorted(('City', 'Technic', 'Star-Wars', 'Creator', 'Ninjago', 'Architecture', 'Duplo', 'Friends', 'DC-Comics-Super-Heroes'))
20
+ learn_color = load_learner('models/lego_convnext_small_4ep_sets05-19.pkl')
21
  learn_gray = load_learner('models/lego_convnext_small_4ep_grayscale.pkl')
22
 
23
  def classify(img, is_color):
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ fastai==2.7.9
2
+ gradio==3.4