Update app.py

app.py

@@ -1,42 +1,10 @@
-from flask import Flask, request, send_file
-from io import BytesIO
-import mimetypes
-
-def allowed_file(filename):
-    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+from flask import Flask
 
 app = Flask(__name__)
 
-ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
-
-
 @app.route('/')
 def hello():
-    return "Hello, Hugging Face!"
-
-@app.route('/1', methods=['POST'])
-def upload_file():
-    if 'image' not in request.files:
-        return {"message": "No file part"}, 400
-    file = request.files['image']
-    if file.filename == '':
-        return {"message": "No selected file"}, 400
-    
-    # send file with angles on image or just angles
-    if file and allowed_file(file.filename):
-        try:
-            file.seek(0)
-            mime_type, _ = mimetypes.guess_type(file.filename)
-            
-            return send_file(
-                BytesIO(file.read()), mimetype=mime_type,
-                as_attachment=False, download_name=file.filename
-            )
-        
-        except Exception as e:
-            return {"message": f"Error processing file: {str(e)}"}, 500
-    return {"message": "File not allowed"}, 400
-    
+    return "Hello from Hugging Face Docker Space!"
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(host='0.0.0.0', port=7860)