AIQuest commited on
Commit
4cf103f
·
verified ·
1 Parent(s): 004faf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('D:/COMSATS/FYP/Coding Part/ModelMaking/Computer_Vision_Part/3rd Try/runs/detect/train9/weights/best.pt') # custome yolo model path
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"),