Spaces:
Sleeping
Sleeping
indentation error fixed
Browse files
app.py
CHANGED
|
@@ -73,8 +73,8 @@ def get_tryon_result(model_name, garment1, garment2, seed=1234):
|
|
| 73 |
# print(model_name)
|
| 74 |
|
| 75 |
if not model_name.startswith("AI Model "):
|
| 76 |
-
|
| 77 |
-
|
| 78 |
|
| 79 |
encoded_garment1 = cv2.imencode('.jpg', garment1)[1].tobytes()
|
| 80 |
encoded_garment1 = base64.b64encode(encoded_garment1).decode('utf-8')
|
|
|
|
| 73 |
# print(model_name)
|
| 74 |
|
| 75 |
if not model_name.startswith("AI Model "):
|
| 76 |
+
model_name = "AI Model " + model_name.split("/")[-1].split(".")[0]
|
| 77 |
+
print(model_name)
|
| 78 |
|
| 79 |
encoded_garment1 = cv2.imencode('.jpg', garment1)[1].tobytes()
|
| 80 |
encoded_garment1 = base64.b64encode(encoded_garment1).decode('utf-8')
|