sleepyml commited on
Commit
8aaebb0
·
1 Parent(s): 3eca1f4

missed an import oops

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -10,6 +10,9 @@ import torchvision
10
  import torchvision.transforms as transforms
11
  import gradio as gr
12
 
 
 
 
13
  # torchvision transforms
14
  normalize = transforms.Normalize(
15
  mean = [0.485, 0.456, 0.406],
@@ -29,6 +32,7 @@ unnormalize = transforms.Compose([
29
 
30
 
31
  # inference script for huggingface space (assumes model is already loaded)
 
32
  def inference(url=None):
33
  # assert url is not None and pil_img is not None
34
  if url:
 
10
  import torchvision.transforms as transforms
11
  import gradio as gr
12
 
13
+ from utils import *
14
+
15
+
16
  # torchvision transforms
17
  normalize = transforms.Normalize(
18
  mean = [0.485, 0.456, 0.406],
 
32
 
33
 
34
  # inference script for huggingface space (assumes model is already loaded)
35
+ # TODO: interface supports drag & drop or url
36
  def inference(url=None):
37
  # assert url is not None and pil_img is not None
38
  if url: