Annonymous commited on
Commit
da85b9a
1 Parent(s): df0b2e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import matplotlib
6
  import matplotlib.pyplot as plt
7
  from PIL import Image
8
  import cv2
 
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
 
11
  from data_transforms import normal_transforms, no_shift_transforms, ig_transforms, modify_transforms
 
6
  import matplotlib.pyplot as plt
7
  from PIL import Image
8
  import cv2
9
+ import gradio as gr
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
 
12
  from data_transforms import normal_transforms, no_shift_transforms, ig_transforms, modify_transforms