app.py

import gradio as gr
import numpy as np
import torch
import cv2
import os

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("device: %s" % device)