Update app.py
Browse files
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)
|