Create app.py

app.py

@@ -0,0 +1,51 @@
+from flask import Flask, request, jsonify
+from PIL import Image
+import base64
+import io
+import numpy as np
+import torch
+import torch.nn.functional as F
+from torchvision.transforms.functional import normalize
+
+
+# Initialize Flask app
+app = Flask(__name__)
+
+# Function to decode a base64 encoded image to a PIL image
+def decode_image_from_base64(image_data):
+    encoded_image = image_data.split(",")[1]
+    decoded_image = base64.b64decode(encoded_image)
+    image = Image.open(io.BytesIO(decoded_image))
+    return image
+
+# Function to encode a PIL image to base64
+def encode_image_to_base64(image):
+    buffered = io.BytesIO()
+    image.save(buffered, format="PNG")
+    encoded_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return "data:image/png;base64," + encoded_image
+
+# Function to process the image
+def process(image_data):
+    image = decode_image_from_base64(image_data)
+    
+
+    return image
+
+@app.route("/")
+def root():
+    return "Welcome to StyleSync Outfit Backround API!"
+
+# Route for the REST API
+@app.route('/api/tryon', methods=['POST'])
+def classify():
+    data = request.json
+    print(data)
+    image_data = data['image']
+    result_image = process(image_data)
+    result_base64 = encode_image_to_base64(result_image)
+    return jsonify({'result': result_base64})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)
+