khan994 commited on
Commit
750a16f
·
1 Parent(s): 5bf5a9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  from fastai.vision.all import *
 
2
  import gradio as gr
3
  import glob
4
 
@@ -26,6 +27,7 @@ class Hook():
26
  def gradcam(img_create):
27
 
28
  pred,idx,probs=learn.predict(img_create)
 
29
  x,= first(dls_augmented.test_dl([img_create]))
30
  hook_output = Hook()
31
  hook = learn.model[0].register_forward_hook(hook_output.hook_func)
 
1
  from fastai.vision.all import *
2
+ import cv2
3
  import gradio as gr
4
  import glob
5
 
 
27
  def gradcam(img_create):
28
 
29
  pred,idx,probs=learn.predict(img_create)
30
+
31
  x,= first(dls_augmented.test_dl([img_create]))
32
  hook_output = Hook()
33
  hook = learn.model[0].register_forward_hook(hook_output.hook_func)