Uploaded 7 changes

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,14 @@
+# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app.main:app"]

README.md

@@ -1,10 +1,10 @@
----
-title: Wearon
-emoji: 🌍
-colorFrom: green
-colorTo: gray
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Wearon
+emoji: 🌍
+colorFrom: green
+colorTo: gray
+sdk: docker
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

default

@@ -0,0 +1,49 @@
+-----BEGIN OPENSSH PRIVATE KEY-----
+b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAACFwAAAAdzc2gtcn
+NhAAAAAwEAAQAAAgEAqJTGiCvjc/h/PvPH0OjruQJIm1Q/keXiDRtZzAESUz3sP+DHfyaq
+dUXbD+CP441+bKGxT3huW0DbWQbyw59QBWeLaE9xvPjTTE2UEXE3VegiJpyVjeW76LKJ88
+zWs7aV11h7vzUTx2D6dwDxTT7UjJw8hef/Z8sDvS5waJNt9pbSXzoqRGHxtrUsrmw2iZtB
+8cdMVsgLgX7k+MHeRQ2ZgxJkbCBJ2s3GdISippnUmcV9Qrkcztg7nsf/88NUrHZASBJxFW
+z3EfikFrJK8putOZSjZBse3SP0v7FslXTNSw3DT9oQnXwcimBpkOU+tJU48xlGDgkAuSDF
+bYWudo6WKcKBwn6HKB1/DwqiHQrw0uzd1sgtM7DLbJsLOaEmhfRI6U+ztc9ckQPdd7T2UP
+/+PzPcnojSgSYIeMLBnF0mhhYpdoS2yNsIge+hi3KzHUMrjd5Ep94VBwlf+NvAvpHm9rse
+0NLoiplR3+Cqo8d4radsAKuZ+SnCntBPPa98+JDwyCWyZmuDFwOagaRL78tJ8uBxIs+YyV
+qAAD2SwfsCQ88M8+tKGlFyAxxT41YdWLdlWbOHA3MTj0WJBPNsn0sSGO5SR+kvRFQN3EYo
+/a90QXFDK+0QxWECmq6eAzQmIkVRRX85EG9eveom2rs638RwBvH6pA8OS0ZvM8Ju6kK3VB
+cAAAdQHz8/Lh8/Py4AAAAHc3NoLXJzYQAAAgEAqJTGiCvjc/h/PvPH0OjruQJIm1Q/keXi
+DRtZzAESUz3sP+DHfyaqdUXbD+CP441+bKGxT3huW0DbWQbyw59QBWeLaE9xvPjTTE2UEX
+E3VegiJpyVjeW76LKJ88zWs7aV11h7vzUTx2D6dwDxTT7UjJw8hef/Z8sDvS5waJNt9pbS
+XzoqRGHxtrUsrmw2iZtB8cdMVsgLgX7k+MHeRQ2ZgxJkbCBJ2s3GdISippnUmcV9Qrkczt
+g7nsf/88NUrHZASBJxFWz3EfikFrJK8putOZSjZBse3SP0v7FslXTNSw3DT9oQnXwcimBp
+kOU+tJU48xlGDgkAuSDFbYWudo6WKcKBwn6HKB1/DwqiHQrw0uzd1sgtM7DLbJsLOaEmhf
+RI6U+ztc9ckQPdd7T2UP/+PzPcnojSgSYIeMLBnF0mhhYpdoS2yNsIge+hi3KzHUMrjd5E
+p94VBwlf+NvAvpHm9rse0NLoiplR3+Cqo8d4radsAKuZ+SnCntBPPa98+JDwyCWyZmuDFw
+OagaRL78tJ8uBxIs+YyVqAAD2SwfsCQ88M8+tKGlFyAxxT41YdWLdlWbOHA3MTj0WJBPNs
+n0sSGO5SR+kvRFQN3EYo/a90QXFDK+0QxWECmq6eAzQmIkVRRX85EG9eveom2rs638RwBv
+H6pA8OS0ZvM8Ju6kK3VBcAAAADAQABAAACADL1vkn9kmVFMhnykW2Ka29izFYAErNk3X9+
+zBdKH6MALgFlTokWRCLtOfpDtNths1zdR/8w9ZMqOORhoFMYPpVTpqWilkw8n4sKTH81TT
+fv0xr+7IqXbsU560/ClQiAYi1VtxqmzdLQcwY0R71FB/V1aSpFZkSW7wtqFVrXbj9DelaE
++6/uN6Y2gEo9tgLE5xtSVrZJBIlhtY+o8yrGoKgpkCfLXugAmkQwsAZSrBzLwUDBOn8giG
+1IIgwq+NalbGsdAAgtB0bsdZXS0asi+itUzkt9k0fgDaVA2gUX6NMOVwM1dR5WHT6LRH4j
+fmASV8PM2Qt3FzvfNy8TYRi/o46AJMKgLWZerVhF3voGckP0mbBwUDZeCpoaP3YBUNmQQ2
+ypszbpvQJ9UIrg3oCWU5V6x5BXAPWjDur1EBpiNBtdxdUXtfzkiUHztp1ih6keoVbAZGPo
+BpOKpDtb1XfMLhcfESSy4VindFjkn3mbD5xIT50AgYTcsErbp+oY5y5OiBNCIBDgS0B87q
+k3+Y94uy0thwFoXENuS7FBa7rx52o00XEb+mguCIxmHXd0QS8ZVjzzDylTti+/9kIU1avl
+gQy5x8QPXRVmH7TRSjwPb0Q6UKk78kXkjxxN+05/ptmvG2RRMSHrwPaQWNsFMJ2H8VfrJY
+1U+rW2DwWpaInF8T+ZAAABAQCw9AUkXs6H/teWS1OCl7Q1A8Tq1He7BQDSOpX48khGp+Qh
+dsTgL8PVSUycb8WMeidVijbrKT1uXA1n+KWjJmRXUfo4K+GQNjFYjeDFl2zy2kmlfI6eDP
+UmzPsY5BURGExzNwj8sj12QL3cPBhvo4L633mSzt7cO6noO2z7ZSDFTozeC22LOU8yREkj
+djrfOi5VlZcZHBx5NdePZe0NmP8PGu9pvD5zFDh2v1g4/br9Vbr2vBkulR3juyplrgcipt
+HJsy1et5TXedjfH5wsshlm8ry9RKBX3iCc9KUcIB7y1wWLNV3R/nR/LHlb1I8aoHweEEUq
+7S8MfpZJ774ksEsMAAABAQDf+T5u1oaVofPhOv1Ww0A4pro4+K2yLh6+sv3EMdVDbFL5Jv
+bEwWPaIfBVQ/i6UjP/IbBDewsA6TVnEvUtYwg/SEE504O6wNsEkaX+DQpPM+KD7FKlSMeH
+MAXR3qHvUON69fjaij89fRShMvHcnao8bNGV5Rd6DK76nbBpf1q4oZSzr60HYWHFMSxrl6
+0Vd/bc2LAH2tPvsAlhpiqRv2oJw7Fl0M1B/JcMk27jyM8L+QSncb4YsxtE/3/NM5SN6f3O
+ha/lgIKAau35RQ66Lk5h7dO2zZvrnVZD4GdShtoGAz4yOQroIG2atXhAheY9agboc9qBPJ
+AX+DhWgBS4+HgVAAABAQDAr9dBMbqPUs0l7G4BoSFdp5XLbT4hpY88UqXxkYcHTD+J5BQy
+4qiakWnDkRce/nNagD/4VbwNrJrhPpLlF+Q1vQcqnsZnss8N/wTNhsSuFChday6yF2vulL
+dMf72VpxOpf0hdzmzyJCg1wh+X59p+oHlE5DKWy3vfUs9ioqsELM4U6mD/zX9UkioyEKXe
+6sh8BvjXqpQEGxYacQnUC3pdGYbIklreUcOw0+HjOPUY67pGc2qb79tWnH09Yee3uuXwYb
+Dzz/ujN0cJW9VMjtJVZgUaHIbjO8avABL1iyh8O0k1fSu6P51fsYnJrb3vtQTgbZiPM446
+fFx/+PnD9Zp7AAAAEzNicGF0aWwyNkBnbWFpbC5jb20BAgMEBQYH
+-----END OPENSSH PRIVATE KEY-----

default.pub

@@ -0,0 +1 @@
+ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAACAQColMaIK+Nz+H8+88fQ6Ou5AkibVD+R5eING1nMARJTPew/4Md/Jqp1RdsP4I/jjX5sobFPeG5bQNtZBvLDn1AFZ4toT3G8+NNMTZQRcTdV6CImnJWN5bvosonzzNaztpXXWHu/NRPHYPp3APFNPtSMnDyF5/9nywO9LnBok232ltJfOipEYfG2tSyubDaJm0Hxx0xWyAuBfuT4wd5FDZmDEmRsIEnazcZ0hKKmmdSZxX1CuRzO2Duex//zw1SsdkBIEnEVbPcR+KQWskrym605lKNkGx7dI/S/sWyVdM1LDcNP2hCdfByKYGmQ5T60lTjzGUYOCQC5IMVtha52jpYpwoHCfocoHX8PCqIdCvDS7N3WyC0zsMtsmws5oSaF9EjpT7O1z1yRA913tPZQ//4/M9yeiNKBJgh4wsGcXSaGFil2hLbI2wiB76GLcrMdQyuN3kSn3hUHCV/428C+keb2ux7Q0uiKmVHf4Kqjx3itp2wAq5n5KcKe0E89r3z4kPDIJbJma4MXA5qBpEvvy0ny4HEiz5jJWoAAPZLB+wJDzwzz60oaUXIDHFPjVh1Yt2VZs4cDcxOPRYkE82yfSxIY7lJH6S9EVA3cRij9r3RBcUMr7RDFYQKarp4DNCYiRVFFfzkQb1696ibauzrfxHAG8fqkDw5LRm8zwm7qQrdUFw== [email protected]

main.py

@@ -0,0 +1,95 @@
+from flask import Flask, request, jsonify, send_from_directory
+from gradio_client import Client, file
+from flask_cors import CORS
+import os
+import traceback
+import shutil
+import base64
+
+app = Flask(__name__)
+CORS(app)
+
+client = Client("yisol/IDM-VTON")
+
+# Directory to save uploaded and processed files
+UPLOAD_FOLDER = 'static/uploads'
+RESULT_FOLDER = 'static/results'
+if not os.path.exists(UPLOAD_FOLDER):
+    os.makedirs(UPLOAD_FOLDER)
+if not os.path.exists(RESULT_FOLDER):
+    os.makedirs(RESULT_FOLDER)
+
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+app.config['RESULT_FOLDER'] = RESULT_FOLDER
+
+@app.route('/process', methods=['POST'])
+def predict():
+    try:
+        # Get the product image URL from the request
+        product_image_url = request.form.get('product_image_url')
+        
+        # Handle the uploaded model image
+        if 'model_image' not in request.files:
+            return jsonify(error='No model image file provided'), 400
+
+        model_image = request.files['model_image']
+        if model_image.filename == '':
+            return jsonify(error='No selected file'), 400
+
+        # Save the uploaded file to the upload directory
+        filename = os.path.join(app.config['UPLOAD_FOLDER'], model_image.filename)
+        model_image.save(filename)
+
+        base_path = os.getcwd()
+        full_filename = os.path.normpath(os.path.join(base_path, filename))
+
+        print("Product image = ", product_image_url)
+        print("Model image = ", full_filename)
+        
+        # Perform prediction
+        try:
+            result = client.predict(
+                dict={"background": file(full_filename), "layers": [], "composite": None},
+                garm_img=file(product_image_url),
+                garment_des="Hello!!",
+                is_checked=True,
+                is_checked_crop=False,
+                denoise_steps=30,
+                seed=42,
+                api_name="/tryon"
+            )
+        except Exception as e:
+            traceback.print_exc()
+            raise
+
+        print(result)
+        # Extract the path of the first output image
+        output_image_path = result[0]
+
+        # Copy the output image to the RESULT_FOLDER
+        output_image_filename = os.path.basename(output_image_path)
+        local_output_path = os.path.join(app.config['RESULT_FOLDER'], output_image_filename)
+        shutil.copy(output_image_path, local_output_path)
+
+        # Remove the uploaded file after processing
+        os.remove(filename)
+
+        # Encode the output image in base64
+        with open(local_output_path, "rb") as image_file:
+            encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
+
+        # Return the output image in JSON format
+        return jsonify(image=encoded_image), 200
+
+    except Exception as e:
+        traceback.print_exc()
+        return jsonify(error=str(e)), 500
+
+@app.route('/uploads/<filename>')
+def uploaded_file(filename):
+    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)
+
+ 

requirements.txt

@@ -0,0 +1,4 @@
+flask
+flask-cors
+gradio_client
+gunicorn