Spaces:
Sleeping
Sleeping
Minor changes to code
Browse files
app.py
CHANGED
|
@@ -8,27 +8,27 @@ from huggingface_hub import InferenceClient
|
|
| 8 |
import os
|
| 9 |
my_key=(os.getenv('my_write_token'))
|
| 10 |
|
| 11 |
-
|
| 12 |
provider="hf-inference",
|
| 13 |
api_key=my_key,
|
| 14 |
)
|
| 15 |
|
| 16 |
# output is a PIL.Image object
|
| 17 |
def generate_image_model_001(prompt):
|
| 18 |
-
image =
|
| 19 |
prompt,
|
| 20 |
model="stabilityai/stable-diffusion-xl-base-1.0",
|
| 21 |
)
|
| 22 |
return image
|
| 23 |
|
| 24 |
# second model definitions here
|
| 25 |
-
|
| 26 |
provider="nebius",
|
| 27 |
api_key=my_key,
|
| 28 |
)
|
| 29 |
|
| 30 |
def generate_image_model_002(prompt):
|
| 31 |
-
image =
|
| 32 |
prompt,
|
| 33 |
model="black-forest-labs/FLUX.1-dev",
|
| 34 |
)
|
|
@@ -38,8 +38,8 @@ my_interface=gr.Interface(
|
|
| 38 |
fn=generate_image_model_002,
|
| 39 |
inputs=gr.Textbox(label="Enter your prompt"),
|
| 40 |
outputs=gr.Image(type="pil", label="Generated Image"),
|
| 41 |
-
title="
|
| 42 |
-
description="Generate images from text prompts
|
| 43 |
)
|
| 44 |
|
| 45 |
my_interface.launch(share=True)
|
|
|
|
| 8 |
import os
|
| 9 |
my_key=(os.getenv('my_write_token'))
|
| 10 |
|
| 11 |
+
first_client = InferenceClient(
|
| 12 |
provider="hf-inference",
|
| 13 |
api_key=my_key,
|
| 14 |
)
|
| 15 |
|
| 16 |
# output is a PIL.Image object
|
| 17 |
def generate_image_model_001(prompt):
|
| 18 |
+
image = first_client.text_to_image(
|
| 19 |
prompt,
|
| 20 |
model="stabilityai/stable-diffusion-xl-base-1.0",
|
| 21 |
)
|
| 22 |
return image
|
| 23 |
|
| 24 |
# second model definitions here
|
| 25 |
+
second_client = InferenceClient(
|
| 26 |
provider="nebius",
|
| 27 |
api_key=my_key,
|
| 28 |
)
|
| 29 |
|
| 30 |
def generate_image_model_002(prompt):
|
| 31 |
+
image = second_client.text_to_image(
|
| 32 |
prompt,
|
| 33 |
model="black-forest-labs/FLUX.1-dev",
|
| 34 |
)
|
|
|
|
| 38 |
fn=generate_image_model_002,
|
| 39 |
inputs=gr.Textbox(label="Enter your prompt"),
|
| 40 |
outputs=gr.Image(type="pil", label="Generated Image"),
|
| 41 |
+
title="Image Generation",
|
| 42 |
+
description="Generate images from text prompts",
|
| 43 |
)
|
| 44 |
|
| 45 |
my_interface.launch(share=True)
|