Spaces:
Runtime error
Runtime error
| import os | |
| import sys | |
| import cv2 | |
| import json | |
| import random | |
| import time | |
| import pickle | |
| import requests | |
| import func_timeout | |
| import numpy as np | |
| import gradio as gr | |
| from PIL import Image | |
| from collections import OrderedDict | |
| taskType = "3" | |
| try: | |
| taskType = str(os.environ['taskType']) | |
| except Exception as e: | |
| print(e) | |
| CapVisible = True | |
| Param4Label = "" | |
| Param4Visible = False | |
| Param5Label = "" | |
| Param5Visible = False | |
| TempInter = True | |
| if taskType=="3": | |
| # 是否显示token输入框 | |
| is_show_token = True | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Poster Background Generation</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "Choose a template" | |
| TempVisible = True | |
| UserLabel = "product photo" | |
| CaptionLabel = "English caption text (optional)" | |
| elif taskType=="4": | |
| # 是否显示token输入框 | |
| TempInter = False | |
| is_show_token = True | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: AI Model Replace</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "Choose a Model" | |
| TempVisible = True | |
| UserLabel = "upload photo" | |
| CaptionLabel = "English caption text (optional)" | |
| elif taskType=="6": | |
| # 是否显示token输入框 | |
| is_show_token = True | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Image Style Filter</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "Choose a Style" | |
| TempVisible = True | |
| UserLabel = "upload photo" | |
| CaptionLabel = "English caption text (optional)" | |
| elif taskType=="5": | |
| # 是否显示token输入框 | |
| is_show_token = True | |
| TempInter = False | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Poster(human) Background Swap</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "background info" | |
| TempVisible = True | |
| UserLabel = "upload photo" | |
| CaptionLabel = "English caption text (optional)" | |
| elif taskType=="1": | |
| # 是否显示token输入框 | |
| is_show_token = True | |
| TempInter = False | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Object Segment</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "guide" | |
| TempVisible = True | |
| UserLabel = "upload photo" | |
| CaptionLabel = "English caption text (optional)" | |
| CapVisible = False | |
| elif taskType=="2": | |
| # 是否显示token输入框 | |
| is_show_token = True | |
| TempInter = False | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Object Remove</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "guide" | |
| TempVisible = True | |
| UserLabel = "upload and paint photo" | |
| CaptionLabel = "English caption text (optional)" | |
| CapVisible = False | |
| elif taskType=="7": | |
| # 是否显示token输入框 | |
| is_show_token = True | |
| TempInter = False | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Image Outpaint</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| # 组件信息 | |
| TempLabel = "guide" | |
| TempVisible = True | |
| UserLabel = "upload and paint photo" | |
| CaptionLabel = "English caption text (required)" | |
| CapVisible = True | |
| Param4Label = "How much to expand width (0 - 2)" | |
| Param4Visible = True | |
| Param5Label = "How much to expand height (0 - 2)" | |
| Param5Visible = True | |
| else: | |
| # 是否显示token输入框 | |
| is_show_token = False | |
| # 标题 | |
| title = r""" | |
| <h1 align="center">Selfit AI App: Poster(product) background generation</h1> | |
| """ | |
| # 标题下的描述 | |
| description = r""" | |
| <a href='https://heybeauty.ai/extension' target='_blank'><b> Go to HeyBeauty for Faster and Free AI App! 🤗 </b></a>.<br> | |
| <b>Official 🤗 Gradio demo</b> for <a href='https://github.com/selfitcamera/Outfit-Anyone-in-the-Wild' target='_blank'><b>Product poster background generation</b></a>.<br> | |
| 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. | |
| 2. <a href='https://heybeauty.ai/keys' target='_blank'><b>You can find your api key here</b></a> .<br> | |
| 3. Share app with your friends and enjoy! 😊.<br> | |
| """ | |
| if is_show_token: | |
| # 当试用次数耗尽，显示的信息 | |
| no_more_attempts = "You've exhausted all trial attempts. [Enter Your API Key](https://heybeauty.ai/keys) to continue..." | |
| else: | |
| # 当试用次数耗尽，显示的信息 | |
| no_more_attempts = "You've exhausted all trial attempts." | |