add load_learner

app.py

@@ -1,8 +1,12 @@
+from fastai.vision.all import *
 from fastapi import FastAPI
 from fastapi.responses import HTMLResponse
 
 app = FastAPI()
 
+# Load the model
+learn = load_learner('model.pkl')
+
 @app.get("/")
 def read_root():
     html_content = "<p>This is a model inference point for the <a href='https://huggingface.co/spaces/vishalbakshi/isitadigit'>isitadigit</a> space</p>"
@@ -12,7 +16,7 @@ if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)
 
-# from fastai.vision.all import *
+
 # from fastapi import FastAPI, HTTPException
 # from pydantic import BaseModel
 # from PIL import Image