Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import torch
|
2 |
import torchvision
|
3 |
from torchvision.models.detection import FasterRCNN
|
4 |
-
from torchvision.models.detection.
|
5 |
from torchvision.transforms import functional as F
|
6 |
from PIL import Image, ImageDraw, ImageFont
|
7 |
import gradio as gr
|
|
|
1 |
import torch
|
2 |
import torchvision
|
3 |
from torchvision.models.detection import FasterRCNN
|
4 |
+
from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
|
5 |
from torchvision.transforms import functional as F
|
6 |
from PIL import Image, ImageDraw, ImageFont
|
7 |
import gradio as gr
|