fix: start processing thread
This commit is contained in:
		| @@ -1,3 +1,4 @@ | ||||
| env | ||||
| .git | ||||
| **/__pycache__ | ||||
| q | ||||
|   | ||||
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1,2 +1,3 @@ | ||||
| env | ||||
| **/__pycache__ | ||||
| q | ||||
|   | ||||
| @@ -13,4 +13,4 @@ RUN apt update && apt install -y ffmpeg libsm6 libxext6 | ||||
| COPY src . | ||||
| COPY best.pt /best.pt | ||||
|  | ||||
| CMD [ "gunicorn", "-b 0.0.0.0:80", "app:app" ] | ||||
| CMD [ "gunicorn", "-b 0.0.0.0:80", "--workers", "1", "app:app" ] | ||||
|   | ||||
| @@ -5,4 +5,4 @@ services: | ||||
|     build: . | ||||
|     restart: unless-stopped | ||||
|     ports: | ||||
|       - 8759:8000 | ||||
|       - 8759:80 | ||||
|   | ||||
| @@ -10,7 +10,7 @@ | ||||
| <img id="img" src="" /> | ||||
|  | ||||
| <p id="statusline">submit an image to get started!</p> | ||||
| <form id="form" action="http://localhost:5000/submit" method="post"> | ||||
| <form id="form" action="http://localhost:8759/submit" method="post"> | ||||
| <input type="file" id="job" name="image" accept="image/png,image/jpeg" /><br> | ||||
| <button type="submit" id="submit">submit</button> | ||||
| </form> | ||||
| @@ -38,11 +38,11 @@ | ||||
|     console.log(id); | ||||
|  | ||||
|     while (true) { | ||||
|       const resp = await fetch('http://localhost:5000/state/' + id); | ||||
|       const resp = await fetch('http://localhost:8759/state/' + id); | ||||
|       const state = (await resp.json()).state; | ||||
|       statusline.innerHTML = state; | ||||
|       if (state == "READY") { | ||||
|         img.src = 'http://localhost:5000/result/' + id | ||||
|         img.src = 'http://localhost:8759/result/' + id | ||||
|         return; | ||||
|       } | ||||
|     } | ||||
|   | ||||
| @@ -44,7 +44,7 @@ def process(): | ||||
|         print("waiting for/getting job") | ||||
|         job = q.get() | ||||
|         filepath = root.joinpath(job) | ||||
|         print("got job") | ||||
|         print("got job", flush=True) | ||||
|         state[job] = { | ||||
|             'state': QUEUE_STATE_PROCESSING, | ||||
|         } | ||||
| @@ -65,6 +65,10 @@ def process(): | ||||
|         last = time.time() | ||||
|  | ||||
|  | ||||
| p = Thread(target=process) | ||||
| p.start() | ||||
|  | ||||
|  | ||||
| @app.route("/submit", methods=['POST']) | ||||
| def render(): | ||||
|     id = str(uuid.uuid4()) | ||||
| @@ -87,7 +91,4 @@ def get_result(i): | ||||
|  | ||||
|  | ||||
| if __name__ == '__main__': | ||||
|     p = Thread(target=process) | ||||
|     p.start() | ||||
|     app.run(debug=True) | ||||
|     p.join() | ||||
|   | ||||
		Reference in New Issue
	
	Block a user