Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import cv2
|
|
2 |
import streamlit as st
|
3 |
import tempfile
|
4 |
import torch
|
|
|
5 |
from torchvision import transforms
|
6 |
from mtcnn import MTCNN
|
7 |
from skimage.feature import hog
|
|
|
2 |
import streamlit as st
|
3 |
import tempfile
|
4 |
import torch
|
5 |
+
import torch.nn as nn
|
6 |
from torchvision import transforms
|
7 |
from mtcnn import MTCNN
|
8 |
from skimage.feature import hog
|