Update myapp.py
Browse files
myapp.py
CHANGED
@@ -12,6 +12,11 @@ CORS(myapp) # Enable CORS for all routes
|
|
12 |
HF_TOKEN = os.environ.get("HF_TOKEN") # Ensure to set your Hugging Face token in the environment
|
13 |
client = InferenceClient(token=HF_TOKEN)
|
14 |
|
|
|
|
|
|
|
|
|
|
|
15 |
# Function to generate an image from a prompt
|
16 |
def generate_image(prompt, seed=1, model="prompthero/openjourney-v4"):
|
17 |
try:
|
|
|
12 |
HF_TOKEN = os.environ.get("HF_TOKEN") # Ensure to set your Hugging Face token in the environment
|
13 |
client = InferenceClient(token=HF_TOKEN)
|
14 |
|
15 |
+
@myapp.route('/')
|
16 |
+
def home():
|
17 |
+
return "Welcome to the Image Background Remover!"
|
18 |
+
|
19 |
+
|
20 |
# Function to generate an image from a prompt
|
21 |
def generate_image(prompt, seed=1, model="prompthero/openjourney-v4"):
|
22 |
try:
|