anon5 commited on
Commit
c55b8a7
1 Parent(s): a8a4d60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify
2
  from transformers import ViTFeatureExtractor, ViTForImageClassification
3
  from PIL import Image
4
  import requests
 
5
 
6
  print("Loading models...")
7
 
@@ -11,6 +12,9 @@ model = ViTForImageClassification.from_pretrained('google/vit-base-patch32-384')
11
  print("Starting webapp...")
12
 
13
  app = Flask(__name__)
 
 
 
14
 
15
  print("Ready")
16
 
 
2
  from transformers import ViTFeatureExtractor, ViTForImageClassification
3
  from PIL import Image
4
  import requests
5
+ import logging
6
 
7
  print("Loading models...")
8
 
 
12
  print("Starting webapp...")
13
 
14
  app = Flask(__name__)
15
+ log = logging.getLogger('werkzeug')
16
+ log.disabled = True
17
+ app.logger.disabled = True
18
 
19
  print("Ready")
20