Spaces:

SudiptaKumar
/

object_detection

Paused

Sudipta Nayak commited on Dec 23, 2023

Commit

9c02022

1 Parent(s): cb6b69f

Issue fix-2

Files changed (2) hide show

app/main.py CHANGED Viewed

@@ -32,30 +32,29 @@ async def root(request: Request):
 @app.post("/detect/", response_class=HTMLResponse)
 async def detect_objects(request: Request, file: UploadFile):
     try:
-        # print('Item:', item)
-        # print('File name:', item.file.filename)
-        # input_file = f"uploads/{item.file.filename}"
-        # output_file = f"static/output/{item.file.filename}"
-        # # Save the uploaded file
-        # with open(input_file, "wb") as file:
-        #     file.write(item.file.file.read())
         print('Detect start')
         # Run YOLOv7 detection and save output
-        # 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])
         print('Detect end')
         # Render HTML using Jinja2Templates
-        # return templates.TemplateResponse(
-        #     "result.html",
-        #     {"request": request, "original_image": f"/static/{item.file.filename}", "output_image": f"/static/output/{item.file.filename}"},
-        # )
-        return templates.TemplateResponse("index.html", {"request": request})
     except Exception as e:
         logging.error(f"Error in /detect endpoint: {str(e)}")

 @app.post("/detect/", response_class=HTMLResponse)
 async def detect_objects(request: Request, file: UploadFile):
     try:
+        print('File name:', file.filename)
+        input_file = f"uploads/{file.filename}"
+        output_file = f"static/output/{file.filename}"
+        # Save the uploaded file
+        with open(input_file, "wb") as file:
+            file.write(file.file.read())
         print('Detect start')
         # Run YOLOv7 detection and save output
+        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])
         print('Detect end')
         # Render HTML using Jinja2Templates
+        return templates.TemplateResponse(
+            "result.html",
+            {"request": request, "original_image": f"/static/{file.filename}", "output_image": f"/static/output/{file.filename}"},
+        )
+        # return templates.TemplateResponse("index.html", {"request": request})
     except Exception as e:
         logging.error(f"Error in /detect endpoint: {str(e)}")

app/templates/index.html CHANGED Viewed

@@ -7,7 +7,7 @@
     <title>Object Detection App</title>
 </head>
 <body>
-    <h1>Welcome to Object Detection App</h1>
     <form action="/detect/" method="post" enctype="multipart/form-data">
         <label for="file">Upload an Image or Video :</label>
         <input type="file" id="file" name="file" accept=".jpg, .jpeg, .png, .mp4" required>

     <title>Object Detection App</title>
 </head>
 <body>
+    <h1>Object Detection App</h1>
     <form action="/detect/" method="post" enctype="multipart/form-data">
         <label for="file">Upload an Image or Video :</label>
         <input type="file" id="file" name="file" accept=".jpg, .jpeg, .png, .mp4" required>