anon5 commited on
Commit
399d742
·
verified ·
1 Parent(s): a8373a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,9 +16,9 @@ pipe = pipeline("image-classification", model="/checkpoint-600")
16
  print("Starting webapp...")
17
 
18
  app = Flask(__name__)
19
- #log = logging.getLogger('werkzeug')
20
- #log.disabled = True
21
- #app.logger.disabled = True
22
 
23
  print("Ready")
24
 
@@ -35,4 +35,4 @@ def classify_image():
35
 
36
  output = pipe(images=[image])
37
 
38
- return jsonify(url=url, result=output)
 
16
  print("Starting webapp...")
17
 
18
  app = Flask(__name__)
19
+ log = logging.getLogger('werkzeug')
20
+ log.disabled = True
21
+ app.logger.disabled = True
22
 
23
  print("Ready")
24
 
 
35
 
36
  output = pipe(images=[image])
37
 
38
+ return jsonify(url=url, result=output[0])