fdg

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 from fastai.vision.all import load_learner
 
 def classify_image_color(img):
-    learn_color = load_learner('lego-bricks-detecter-model-color.pkl')
+    learn_color = load_learner('model-color.pkl')
     pred,_,probs = learn_color.predict(img)
     return dict(zip(learn_color.dls.vocab, map(float,probs)))
 
 def classify_image_shape(img):
-    learn_shape = load_learner('lego-bricks-detecter-model-shape.pkl')
+    learn_shape = load_learner('bricks-model.pkl')
     pred,_,probs = learn_shape.predict(img)
     return dict(zip(learn_shape.dls.vocab, map(float,probs)))