Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import numpy as np
|
|
7 |
# function which is returning the number of object detected
|
8 |
def number_object_detected(image):
|
9 |
|
10 |
-
custom_model = YOLO('
|
11 |
results = custom_model(image,verbose= False)
|
12 |
|
13 |
dic = results[0].names
|
@@ -112,7 +112,7 @@ def process_data(files,car_brand, car_name, model_year, mileage, city_registered
|
|
112 |
|
113 |
|
114 |
years_list = list(range(2024, 1899, -1))
|
115 |
-
gr.Interface(fn = process_data,
|
116 |
inputs=[gr.Gallery(label="Upload Files", type="pil"),
|
117 |
gr.Dropdown(['suzuki','toyota','honda','kia'], label='Brand'),
|
118 |
gr.Textbox(lines=1, label="Car Name"),
|
|
|
7 |
# function which is returning the number of object detected
|
8 |
def number_object_detected(image):
|
9 |
|
10 |
+
custom_model = YOLO('best.pt') # custome yolo model path
|
11 |
results = custom_model(image,verbose= False)
|
12 |
|
13 |
dic = results[0].names
|
|
|
112 |
|
113 |
|
114 |
years_list = list(range(2024, 1899, -1))
|
115 |
+
gr.Interface(fn = process_data,title = "Car Price Prediction",
|
116 |
inputs=[gr.Gallery(label="Upload Files", type="pil"),
|
117 |
gr.Dropdown(['suzuki','toyota','honda','kia'], label='Brand'),
|
118 |
gr.Textbox(lines=1, label="Car Name"),
|