Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
import os
|
2 |
-
import cv2
|
3 |
import gradio as gr
|
4 |
from huggingface_hub import hf_hub_download
|
5 |
import onnxruntime as ort
|
@@ -7,6 +6,7 @@ import cv2
|
|
7 |
import numpy as np
|
8 |
from facenet_pytorch import MTCNN
|
9 |
from torchvision import transforms
|
|
|
10 |
import torch
|
11 |
|
12 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
|
|
1 |
import os
|
|
|
2 |
import gradio as gr
|
3 |
from huggingface_hub import hf_hub_download
|
4 |
import onnxruntime as ort
|
|
|
6 |
import numpy as np
|
7 |
from facenet_pytorch import MTCNN
|
8 |
from torchvision import transforms
|
9 |
+
import cv2
|
10 |
import torch
|
11 |
|
12 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|