Hi again!
I’ve tried removing line 11 and replacing lien 5 with from fastai.vision.all import *, but still no cigar. I believe that you may be right about incorrect modules. I am using fastai==2.1.5 and it seems that I should install fastcore along with that. I did some debugging to see where the problem is coming form it has to do with the fastai PyPi version (2.1.5) and torch/torchvision version (1.7.0 and 0.8.0) respectively.
I’ve read some of your other solutions on this thread and may try to deploy wit Heroku and attempt to connect that with my Flutter mobile UI.
Here is my most recent requirements.txt:
aiofiles==0.5.0
aiohttp==3.6.2
asyncio==3.4.3
fastai==2.2.5
fastcore
ipywidgets
graphviz
numpy==1.19.0
torch==1.7.0
torchvision==0.8.0
pillow==8.0.1
python-multipart==0.0.5
starlette==0.13.6
uvicorn==0.11.7
Here is my server.py file in the app directory:
import aiohttp
import asyncio
import uvicorn
from fastai import *
from fastai.vision.all import *
from io import BytesIO
from starlette.applications import Starlette
from starlette.middleware.cors import CORSMiddleware
from starlette.responses import HTMLResponse, JSONResponse
from starlette.staticfiles import StaticFiles
export_file_url = 'https://www.dropbox.com/s/yqmtjsqednhljqt/export.pkl?raw=1'
export_file_name = 'export.pkl'
classes = ['black', 'grizzly', 'teddys']
path = Path(__file__).parent
app = Starlette()
app.add_middleware(CORSMiddleware, allow_origins=['*'], allow_headers=['X-Requested-With', 'Content-Type'])
app.mount('/static', StaticFiles(directory='app/static'))
async def download_file(url, dest):
if dest.exists(): return
async with aiohttp.ClientSession() as session:
async with session.get(url) as response:
data = await response.read()
with open(dest, 'wb') as f:
f.write(data)
async def setup_learner():
await download_file(export_file_url, path / export_file_name)
try:
learn = load_learner(path, export_file_name)
return learn
except RuntimeError as e:
if len(e.args) > 0 and 'CPU-only machine' in e.args[0]:
print(e)
message = "\n\nThis model was trained with an old version of fastai and will not work in a CPU environment.\n\nPlease update the fastai library in your training environment and export your model again.\n\nSee instructions for 'Returning to work' at https://course.fast.ai."
raise RuntimeError(message)
else:
raise
loop = asyncio.get_event_loop()
tasks = [asyncio.ensure_future(setup_learner())]
learn = loop.run_until_complete(asyncio.gather(*tasks))[0]
loop.close()
@app.route('/')
async def homepage(request):
html_file = path / 'view' / 'index.html'
return HTMLResponse(html_file.open().read())
@app.route('/analyze', methods=['POST'])
async def analyze(request):
img_data = await request.form()
img_bytes = await (img_data['file'].read())
img = open_image(BytesIO(img_bytes))
prediction = learn.predict(img)[0]
return JSONResponse({'result': str(prediction)})
if __name__ == '__main__':
if 'serve' in sys.argv:
uvicorn.run(app=app, host='0.0.0.0', port=5000, log_level="info")
As always, any tips, tricks, or learning experience would be greatly appreciated.
Thank you
Ty