KiKi commited on
Commit
cf95d64
·
1 Parent(s): 72a7eba

adding requirements file and dependencies

Browse files
Files changed (2) hide show
  1. app.py +9 -0
  2. requirements.txt +4 -0
app.py CHANGED
@@ -1,6 +1,15 @@
1
  import streamlit as st
2
  import base64
3
  from predictor import predict
 
 
 
 
 
 
 
 
 
4
 
5
  def add_bg_from_local(image_file):
6
  with open(image_file, "rb") as image_file:
 
1
  import streamlit as st
2
  import base64
3
  from predictor import predict
4
+ import torch
5
+ import numpy as np
6
+ import matplotlib.pyplot as plt
7
+ import torchvision.models as models
8
+ import os
9
+ from PIL import Image
10
+ from IPython.display import display
11
+ from torchvision import datasets, transforms
12
+ from torch.utils.data import DataLoader, Subset
13
 
14
  def add_bg_from_local(image_file):
15
  with open(image_file, "rb") as image_file:
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ torch>=1.7.0
2
+ torchvision>=0.8.1
3
+ matplotlib>=3.2.2
4
+ numpy>=1.18.5