Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from pytorch_grad_cam import GradCAM
|
|
| 8 |
from pytorch_grad_cam.utils.image import show_cam_on_image
|
| 9 |
import gradio as gr
|
| 10 |
from huggingface_hub import hf_hub_download
|
| 11 |
-
from
|
| 12 |
|
| 13 |
model = Model('best.pt')
|
| 14 |
|
|
|
|
| 8 |
from pytorch_grad_cam.utils.image import show_cam_on_image
|
| 9 |
import gradio as gr
|
| 10 |
from huggingface_hub import hf_hub_download
|
| 11 |
+
from yolo import Model
|
| 12 |
|
| 13 |
model = Model('best.pt')
|
| 14 |
|