Reality123b commited on
Commit
1d010bc
·
verified ·
1 Parent(s): 756a49f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,10 +1,12 @@
1
  from config import Flask,pipeline_dict,Response,convHandler
2
  from application.chat_inference import ChatInference
3
  from flask import render_template,request
 
4
 
5
  app = Flask(__name__, template_folder='application/templates', static_folder='application/static')
6
 
7
  chat_inference = ChatInference()
 
8
 
9
  @app.route('/')
10
  def home():
@@ -42,5 +44,4 @@ def models():
42
  return list(pipeline_dict['api']['models'].keys())
43
 
44
  if __name__ == "__main__":
45
- app.run(host="0.0.0.0", port=7860, debug=False)
46
-
 
1
  from config import Flask,pipeline_dict,Response,convHandler
2
  from application.chat_inference import ChatInference
3
  from flask import render_template,request
4
+ from application.utils.image_captioning import ImageCaptioning
5
 
6
  app = Flask(__name__, template_folder='application/templates', static_folder='application/static')
7
 
8
  chat_inference = ChatInference()
9
+ image_captioning = ImageCaptioning()
10
 
11
  @app.route('/')
12
  def home():
 
44
  return list(pipeline_dict['api']['models'].keys())
45
 
46
  if __name__ == "__main__":
47
+ app.run(host="0.0.0.0", port=7860, debug=False)