Update app.py
#1
by
Enmartz
- opened
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import torch
|
|
| 4 |
from PIL import Image
|
| 5 |
import numpy as np
|
| 6 |
|
|
|
|
| 7 |
# Cargar el modelo YOLO (usando YOLOv5 como ejemplo)
|
| 8 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # Puedes cambiar 'yolov5s' por cualquier otro modelo
|
| 9 |
|
|
|
|
| 4 |
from PIL import Image
|
| 5 |
import numpy as np
|
| 6 |
|
| 7 |
+
|
| 8 |
# Cargar el modelo YOLO (usando YOLOv5 como ejemplo)
|
| 9 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # Puedes cambiar 'yolov5s' por cualquier otro modelo
|
| 10 |
|