Geek7 commited on
Commit
a40654e
1 Parent(s): ec20299

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +5 -0
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: