Sudipta Nayak commited on
Commit
3ebd9c2
·
1 Parent(s): 695d423

print stsmt added

Browse files
Files changed (1) hide show
  1. app/main.py +6 -0
app/main.py CHANGED
@@ -27,6 +27,8 @@ async def root(request: Request):
27
 
28
  @app.post("/detect", response_class=HTMLResponse)
29
  async def detect_objects(request: Request, item: Item):
 
 
30
  input_file = f"uploads/{item.file.filename}"
31
  output_file = f"static/output/{item.file.filename}"
32
 
@@ -34,9 +36,13 @@ async def detect_objects(request: Request, item: Item):
34
  with open(input_file, "wb") as file:
35
  file.write(item.file.file.read())
36
 
 
 
37
  # Run YOLOv7 detection and save output
38
  subprocess.run(["python", "detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt", "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--project", output_file])
39
 
 
 
40
  # Render HTML using Jinja2Templates
41
  return templates.TemplateResponse(
42
  "result.html",
 
27
 
28
  @app.post("/detect", response_class=HTMLResponse)
29
  async def detect_objects(request: Request, item: Item):
30
+ print('File name:', item.file.filename)
31
+
32
  input_file = f"uploads/{item.file.filename}"
33
  output_file = f"static/output/{item.file.filename}"
34
 
 
36
  with open(input_file, "wb") as file:
37
  file.write(item.file.file.read())
38
 
39
+ print('Detect start')
40
+
41
  # Run YOLOv7 detection and save output
42
  subprocess.run(["python", "detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt", "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--project", output_file])
43
 
44
+ print('Detect end')
45
+
46
  # Render HTML using Jinja2Templates
47
  return templates.TemplateResponse(
48
  "result.html",