Minor updates

app.py

@@ -3,31 +3,20 @@ import platform
 if platform.system()=='Linux':
     pathlib.WindowsPath=pathlib.PosixPath
 
-from fastai.vision.all import *
+import dill
 import gradio as gr
 
-# Add logging for debugging
-import logging
-logging.basicConfig(level=logging.INFO)
+# Load with dill instead of load_learner
+with open('model_dill.pkl','rb') as f:
+    learn=dill.load(f)
 
-try:
-    learn=load_learner('model_dill.pkl')
-    labels=learn.dls.vocab
-    logging.info(f"Model loaded. Classes: {labels}")
-except Exception as e:
-    logging.error(f"Model loading failed: {e}")
-    raise
+labels=learn.dls.vocab
 
 def predict(img):
-    try:
-        img=PILImage.create(img)
-        pred,pred_idx,probs=learn.predict(img)
-        result={labels[i]:float(probs[i]) for i in range(len(labels))}
-        logging.info(f"Prediction: {pred}, Probs: {probs}")
-        return result
-    except Exception as e:
-        logging.error(f"Prediction failed: {e}")
-        return {"Error":"Prediction failed"}
+    from fastai.vision.all import PILImage
+    img=PILImage.create(img)
+    pred,pred_idx,probs=learn.predict(img)
+    return {labels[i]:float(probs[i]) for i in range(len(labels))}
 
 gr.Interface(
     fn=predict,

requirements.txt

@@ -1,2 +1,3 @@
 fastai<2.8.0
-fasttransform
+fasttransform
+dill