simple-predictor / src /mainrun.py
Thiago Andrade
fix docker file and main startup
2b24cc6
from fastapi import FastAPI, Request
from fastcore.transform import Transform
from fastapi.responses import HTMLResponse
from fastapi.middleware.cors import CORSMiddleware
from fastbook import *
from pydantic import BaseModel, Field
import urllib.request
import base64
from fastai.vision.widgets import *
from fastai.vision import *
from fastai.vision.all import *
from fastai.metrics import *
from fastai.data.external import *
from fastai.vision.all import *
import torchvision.transforms as transforms
import os
import pathlib
import base64
from pathlib import *
import PIL
from PIL import Image
from PIL import Image, ImageOps
from fastai.vision.all import *
import torchvision.transforms as transforms
import uvicorn
import asyncio
class GrayscaleTransform(Transform):
def __init__(self):
pass
def encodes(self, img: PILImage):
return PIL.ImageOps.grayscale(img)
def get_fingers(fingers_path):
print('fingers path', fingers_path)
return []
def get_finger_label(finger_path: Path):
print('finger path', finger_path)
return ''
async def main():
server = uvicorn.Server(uvicorn.Config("server:app", port=8080, log_level="debug", host="0.0.0.0"))
# server = uvicorn.Server(uvicorn.Config("server:app", port=7860, log_level="debug", host="0.0.0.0"))
await server.serve()
if __name__ == "__main__":
print('Hello World!!')
asyncio.run(main())