Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,13 @@ from PIL import Image
|
|
| 7 |
import requests
|
| 8 |
import json
|
| 9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
model = YOLO("best.pt")
|
| 11 |
|
| 12 |
GEMINI_API_KEY = "AIzaSyCBs4TumAonKI0AodIzbl4b8Vmu9eM_r9I"
|
|
|
|
| 7 |
import requests
|
| 8 |
import json
|
| 9 |
|
| 10 |
+
# Download sample images
|
| 11 |
+
torch.hub.download_url_to_file('https://github.com/Janno1402/Helmet-License-Plate-Detection/blob/main/Sample-Image-1.jpg?raw=true', 'sample_1.jpg')
|
| 12 |
+
torch.hub.download_url_to_file('https://github.com/Janno1402/Helmet-License-Plate-Detection/blob/main/Sample-Image-2.jpg?raw=true', 'sample_2.jpg')
|
| 13 |
+
torch.hub.download_url_to_file('https://github.com/Janno1402/Helmet-License-Plate-Detection/blob/main/Sample-Image-3.jpg?raw=true', 'sample_3.jpg')
|
| 14 |
+
torch.hub.download_url_to_file('https://github.com/Janno1402/Helmet-License-Plate-Detection/blob/main/Sample-Image-4.jpg?raw=true', 'sample_4.jpg')
|
| 15 |
+
torch.hub.download_url_to_file('https://github.com/Janno1402/Helmet-License-Plate-Detection/blob/main/Sample-Image-5.jpg?raw=true', 'sample_5.jpg')
|
| 16 |
+
|
| 17 |
model = YOLO("best.pt")
|
| 18 |
|
| 19 |
GEMINI_API_KEY = "AIzaSyCBs4TumAonKI0AodIzbl4b8Vmu9eM_r9I"
|