Skip to content

Instantly share code, notes, and snippets.

View mengwangk's full-sized avatar

Meng Wang mengwangk

View GitHub Profile
@mengwangk
mengwangk / main.py
Last active December 29, 2020 07:41
main.py - Backend servince ML Models
import io
import json
import uuid
import uvicorn
from fastapi import FastAPI, File, UploadFile
from PIL import Image
from starlette.responses import Response
from model.dcgan import dcgan
@mengwangk
mengwangk / pgan.py
Created December 29, 2020 09:50
Serving ML models - PGAN front end
import io
import requests
import streamlit as st
from PIL import Image
server_url=f"http://backend:8088/pgan"
def pgan():
st.title("PGAN - Progressive Growing of GANS")
@mengwangk
mengwangk / pgan.py
Created December 29, 2020 09:52
Serving ML Models - backend PGAN
import torch
import torchvision.transforms as Transforms
use_gpu = True if torch.cuda.is_available() else False
# trained on high-quality celebrity faces "celebA" dataset
# this model outputs 512 x 512 pixel images
model = torch.hub.load('facebookresearch/pytorch_GAN_zoo:hub',
'PGAN', model_name='celebAHQ-512',
pretrained=True, useGPU=use_gpu)
@mengwangk
mengwangk / dcgan.py
Created December 29, 2020 10:07
Serving ML Models - Streamlit Front End - DCGAN
import io
import requests
import streamlit as st
from PIL import Image
server_url=f"http://backend:8088/dcgan"
def dcgan():
@mengwangk
mengwangk / dcgan.py
Created December 29, 2020 10:08
Serving ML Models - FastAPI Backend - dcgan.py
import torch
import torchvision.transforms as Transforms
use_gpu = True if torch.cuda.is_available() else False
model = torch.hub.load('facebookresearch/pytorch_GAN_zoo:hub', 'DCGAN', pretrained=True, useGPU=use_gpu)
num_images = 1
def dcgan():
@mengwangk
mengwangk / resnext.py
Created December 29, 2020 10:13
Serving ML Models - Streamlit Front End - resnext.py
import io
import requests
import streamlit as st
from PIL import Image
server_url=f"http://backend:8088/resnext"
def resnext():
st.title("RESNEXT - Next generation ResNets, more efficient and accurate")
@mengwangk
mengwangk / resnext.py
Created December 29, 2020 10:15
Serving ML Models - FastAPI resnext.py
import torch
import torchvision.transforms as Transforms
# Read the categories
with open("imagenet_classes.txt", "r") as f:
categories = [s.strip() for s in f.readlines()]
model = torch.hub.load('pytorch/vision:v0.6.0', 'resnext50_32x4d', pretrained=True)
# or
# model = torch.hub.load('pytorch/vision:v0.6.0', 'resnext101_32x8d', pretrained=True)
@mengwangk
mengwangk / start.sh
Created December 29, 2020 10:24
Serving ML Models - Front End start.sh
#!/usr/bin/env bash
set -e
if [ "$DEBUG" = true ] ; then
echo 'Debugging - ON'
nodemon --exec streamlit run main.py
else
echo 'Debugging - OFF'
streamlit run main.py
fi
@mengwangk
mengwangk / start.sh
Created December 29, 2020 10:27
Serving ML Models - Backend start.sh
#!/usr/bin/env bash
set -e
if [ "$DEBUG" = true ] ; then
echo 'Debugging - ON'
uvicorn main:app --host 0.0.0.0 --port 8088 --reload
else
echo 'Debugging - OFF'
uvicorn main:app --host 0.0.0.0 --port 8088
fi
@mengwangk
mengwangk / rust.vim
Created December 31, 2020 08:06
Rust syntax file
scriptencoding utf-8
if exists('g:no_vim_fancy_text') || !has('conceal') || &enc != 'utf-8'
finish
endif
syntax match rsFancyOperator "!=" conceal cchar=≠
syntax match rsFancyOperator "<=" conceal cchar=≤
syntax match rsFancyOperator ">=" conceal cchar=≥
syntax match rsFancyOperator "<<" conceal cchar=«