Spaces:

Morehod
/

first_model

Sleeping

Jambulat Soltomuradow commited on Mar 10

Commit

9dc6159

1 Parent(s): 674042d

cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,11 @@
 from fastai.vision.all import *
 import gradio as gr
-import numpy as np
-learn = load_learner('model.pkl')
 categories = ('damaged', 'intact')

 from fastai.vision.all import *
 import gradio as gr
+import os
+os.environ["OMP_NUM_THREADS"] = "1"
+os.environ["MKL_NUM_THREADS"] = "1"
+learn = load_learner('model.pkl', cpu=True)
 categories = ('damaged', 'intact')