Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import cv2
|
|
| 4 |
import pandas as pd
|
| 5 |
import smtplib
|
| 6 |
from datetime import datetime, timedelta
|
|
|
|
| 7 |
|
| 8 |
# Load pre-trained model
|
| 9 |
model_name = "hustvl/yolos-small"
|
|
@@ -27,10 +28,14 @@ def detect_number_plates(image):
|
|
| 27 |
|
| 28 |
return number_plates
|
| 29 |
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
# Create a pandas DataFrame
|
| 32 |
df = pd.DataFrame({
|
| 33 |
-
"Date": [datetime.now().strftime("%Y-%m-%d %H:%M:%S") for _ in range(len(number_plates))],
|
| 34 |
"Number Plate": [f"{x1}, {y1}, {x2}, {y2}" for x1, y1, x2, y2 in number_plates]
|
| 35 |
})
|
| 36 |
|
|
@@ -41,6 +46,7 @@ def detect_number_plates(image):
|
|
| 41 |
combined_df.to_excel("number_plates.xlsx", index=False)
|
| 42 |
except FileNotFoundError:
|
| 43 |
df.to_excel("number_plates.xlsx", index=False)
|
|
|
|
| 44 |
def monitor_vehicles(number_plates):
|
| 45 |
# Load registered number plates from a file or database
|
| 46 |
registered_plates = pd.read_csv("registered_plates.csv")["Number Plate"].tolist()
|
|
|
|
| 4 |
import pandas as pd
|
| 5 |
import smtplib
|
| 6 |
from datetime import datetime, timedelta
|
| 7 |
+
import datetime
|
| 8 |
|
| 9 |
# Load pre-trained model
|
| 10 |
model_name = "hustvl/yolos-small"
|
|
|
|
| 28 |
|
| 29 |
return number_plates
|
| 30 |
|
| 31 |
+
def detect_number_plates(image):
|
| 32 |
+
# Your code here
|
| 33 |
+
pass
|
| 34 |
+
|
| 35 |
+
def save_to_excel(number_plates):
|
| 36 |
# Create a pandas DataFrame
|
| 37 |
df = pd.DataFrame({
|
| 38 |
+
"Date": [datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") for _ in range(len(number_plates))],
|
| 39 |
"Number Plate": [f"{x1}, {y1}, {x2}, {y2}" for x1, y1, x2, y2 in number_plates]
|
| 40 |
})
|
| 41 |
|
|
|
|
| 46 |
combined_df.to_excel("number_plates.xlsx", index=False)
|
| 47 |
except FileNotFoundError:
|
| 48 |
df.to_excel("number_plates.xlsx", index=False)
|
| 49 |
+
|
| 50 |
def monitor_vehicles(number_plates):
|
| 51 |
# Load registered number plates from a file or database
|
| 52 |
registered_plates = pd.read_csv("registered_plates.csv")["Number Plate"].tolist()
|