ehsanhon commited on
Commit
1b13910
·
verified ·
1 Parent(s): 9818ab5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -2,9 +2,14 @@ from flask import Flask, request, send_file
2
  from io import BytesIO
3
  import mimetypes
4
 
 
 
5
 
6
  app = Flask(__name__)
7
 
 
 
 
8
  @app.route('/')
9
  def hello():
10
  return "Hello, Hugging Face!"
 
2
  from io import BytesIO
3
  import mimetypes
4
 
5
+ def allowed_file(filename):
6
+ return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
7
 
8
  app = Flask(__name__)
9
 
10
+ ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
11
+
12
+
13
  @app.route('/')
14
  def hello():
15
  return "Hello, Hugging Face!"