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">AI Poster Background Generation</h1> | |
""" | |
# 标题下的描述 | |
description = r""" | |
Go to <a href='https://heybeauty.ai/product-background-swap' 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/home/apikeys' 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">Model Head Swap</h1> | |
""" | |
# 标题下的描述 | |
description = r""" | |
Go to <a href='https://heybeauty.ai/model-head-swap' 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/home/apikeys' 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/home/apikeys' 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/home/apikeys' 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/home/apikeys' 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">AI Object Removal</h1> | |
""" | |
# 标题下的描述 | |
description = r""" | |
Go to <a href='https://heybeauty.ai/remove-object' target='_blank'><b>HeyBeauty</b></a> for Faster and Free Object Removal! 🤗 .<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/home/apikeys' 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/home/apikeys' 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 | |
elif taskType=="9": | |
# 是否显示token输入框 | |
is_show_token = True | |
TempInter = False | |
# 标题 | |
title = r""" | |
<h1 align="center">Red Panda Image Generator by Recraft</h1> | |
""" | |
# 标题下的描述 | |
description = r""" | |
Go to <a href='https://heybeauty.ai/red-panda-recraft' target='_blank'><b>Recraft</b></a> for Faster and Free Generation! 🤗 .<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/home/apikeys' 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 = "caption text (required, any language)" | |
CapVisible = 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/home/apikeys' 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/home/apikeys) to continue..." | |
else: | |
# 当试用次数耗尽,显示的信息 | |
no_more_attempts = "You've exhausted all trial attempts." | |