Update app.py

app.py

@@ -2,14 +2,21 @@ import gradio as gr
 import traceback
 from fastai.vision.all import *
 from huggingface_hub import from_pretrained_fastai
+import torch, os
 
+os.environ.setdefault("OMP_NUM_THREADS", "1")
+torch.set_num_threads(1)
 
 learn = from_pretrained_fastai("Pablogps/castle-classifier-25")
 try:
     learn.to_fp32()
 except Exception:
     pass
-    
+
+for dl in learn.dls.loaders:
+    try: dl.num_workers = 0
+    except: pass
+
 labels = learn.dls.vocab