mayurKuchha commited on
Commit
b07e559
1 Parent(s): 4c64009

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -13
main.py CHANGED
@@ -2,7 +2,6 @@ import io
2
  import os
3
  import sys
4
  import base64
5
- #sys.path.append("/home/dell/RTMDET_VSCODE")
6
  from fastapi import FastAPI, File, UploadFile, HTTPException, Form
7
  from fastapi.responses import HTMLResponse, StreamingResponse
8
  from fastapi.staticfiles import StaticFiles
@@ -18,18 +17,10 @@ from mmdet.apis import inference_detector
18
  # FastAPI instance
19
  app = FastAPI()
20
 
21
- # Set up static files and templates directory
22
- #app.mount("/static", StaticFiles(directory="templates"), name="static")
23
- #templates = Jinja2Templates(directory="templates")
24
-
25
  # Load your models
26
  detection_model = load_detection_model()
27
  ocr_model = load_ocr_model()
28
 
29
- #@app.get("/", response_class=HTMLResponse)
30
- #async def read_root(request: Request):
31
- # return templates.TemplateResponse("upload_form.html", {"request": request})
32
-
33
  @app.get("/", response_class=HTMLResponse)
34
  async def read_root():
35
  return """
@@ -133,7 +124,4 @@ async def create_upload_file(file: UploadFile = File(...), return_type: str = Fo
133
  except Exception as e:
134
  raise HTTPException(status_code=500, detail=str(e))
135
 
136
- # Run the app
137
- #if __name__ == "__main__":
138
- # import uvicorn
139
- # uvicorn.run(app, host="0.0.0.0", port=8000)
 
2
  import os
3
  import sys
4
  import base64
 
5
  from fastapi import FastAPI, File, UploadFile, HTTPException, Form
6
  from fastapi.responses import HTMLResponse, StreamingResponse
7
  from fastapi.staticfiles import StaticFiles
 
17
  # FastAPI instance
18
  app = FastAPI()
19
 
 
 
 
 
20
  # Load your models
21
  detection_model = load_detection_model()
22
  ocr_model = load_ocr_model()
23
 
 
 
 
 
24
  @app.get("/", response_class=HTMLResponse)
25
  async def read_root():
26
  return """
 
124
  except Exception as e:
125
  raise HTTPException(status_code=500, detail=str(e))
126
 
127
+