Spaces:

moshe742
/

chess-or-not

Sleeping

moshe742 commited on Oct 25, 2024

Commit

f3ef851

1 Parent(s): 7bac702

refactor a bit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,16 +1,15 @@
-from fastai.vision.all import *
 from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 categories = ('chess', 'checkers')
 repo_id = "moshe742/chess-or-not"
-learner = from_pretrained_fastai(repo_id)
 def classify_image(image):
-    global learner
-    pred, idx, probs = learner.predict(image)
     return dict(zip(categories, map(float, probs)))

 from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 categories = ('chess', 'checkers')
 repo_id = "moshe742/chess-or-not"
+learn = from_pretrained_fastai(repo_id)
 def classify_image(image):
+    global learn
+    pred, idx, probs = learn.predict(image)
     return dict(zip(categories, map(float, probs)))