fix: start processing thread
This commit is contained in:
parent
2b844a899f
commit
109e06a5d6
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user