test

main.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 import requests
 from torchvision import transforms
+import os
 
 torch.hub._validate_not_a_forked_repo=lambda a,b,c: True