mohamed12ahmed commited on
Commit
bac2bc4
·
verified ·
1 Parent(s): 0e6577c

Update inference_gradio.py

Browse files
Files changed (1) hide show
  1. inference_gradio.py +1 -1
inference_gradio.py CHANGED
@@ -2,7 +2,7 @@ import sys
2
  import cv2
3
  import utils
4
  import numpy as np
5
- import torcth
6
  from PIL import Image
7
  from utils import convert_state_dict
8
  from models import restormer_arch
 
2
  import cv2
3
  import utils
4
  import numpy as np
5
+ import torch
6
  from PIL import Image
7
  from utils import convert_state_dict
8
  from models import restormer_arch