paugar commited on
Commit
60cdc75
·
verified ·
1 Parent(s): ba8ba93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -30,12 +30,7 @@ def process_img(image):
30
  with torch.no_grad():
31
  result = model(y)
32
 
33
-
34
- restored_img = result.cpu().detach().numpy()
35
- restored_img = np.clip(restored_img, 0. , 1.)
36
-
37
- restored_img = (restored_img * 255.0).round().astype(np.uint8) # float32 to uint8
38
- return Image.fromarray(restored_img)
39
 
40
  title = "Efficient Hazy Vehicle Detection ✏️[] 🤗"
41
  description = ''' ## [Efficient Hazy Vehicle Detection](https://github.com/cidautai)
 
30
  with torch.no_grad():
31
  result = model(y)
32
 
33
+ return Image.fromarray(result)
 
 
 
 
 
34
 
35
  title = "Efficient Hazy Vehicle Detection ✏️[] 🤗"
36
  description = ''' ## [Efficient Hazy Vehicle Detection](https://github.com/cidautai)