xyplon commited on
Commit
cf37cf5
·
verified ·
1 Parent(s): aded4bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -14,15 +14,19 @@ limiter = Limiter(
14
  app=app,
15
  default_limits=["8 per minute"]
16
  )
17
-
18
  @app.route("/")
19
  def index():
20
  return render_template('models.html')
21
 
22
-
 
 
23
  @app.route('/gen', methods=['POST'])
24
  @limiter.limit("4 per minute")
25
  def Hf():
 
 
26
  prompt = request.json.get('prompt', '')
27
  negative = request.json.get('negative', '')
28
  steps = request.json.get('steps', 20)
@@ -48,6 +52,7 @@ def Hf():
48
  return "an error occurred! ", 500
49
  for chunk in req.iter_lines():
50
  yield f'{chunk.decode()}\n'
 
51
  return Response(Gen(prompt=prompt,negative=negative,steps=steps,width=width,height=height,scale=scale,style=style,model=model), mimetype="text/event-stream")
52
 
53
 
 
14
  app=app,
15
  default_limits=["8 per minute"]
16
  )
17
+ reqs = 0
18
  @app.route("/")
19
  def index():
20
  return render_template('models.html')
21
 
22
+ @app.route("/info")
23
+ def info():
24
+ return {"pending requests : " : reqs}
25
  @app.route('/gen', methods=['POST'])
26
  @limiter.limit("4 per minute")
27
  def Hf():
28
+ global reqs
29
+ reqs+=1
30
  prompt = request.json.get('prompt', '')
31
  negative = request.json.get('negative', '')
32
  steps = request.json.get('steps', 20)
 
52
  return "an error occurred! ", 500
53
  for chunk in req.iter_lines():
54
  yield f'{chunk.decode()}\n'
55
+ reqs-=1
56
  return Response(Gen(prompt=prompt,negative=negative,steps=steps,width=width,height=height,scale=scale,style=style,model=model), mimetype="text/event-stream")
57
 
58