xeeshan commited on
Commit
bce0bda
·
1 Parent(s): 5fc77e2

Minor changes to code

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- hf_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 = hf_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
- together_client = InferenceClient(
26
  provider="nebius",
27
  api_key=my_key,
28
  )
29
 
30
  def generate_image_model_002(prompt):
31
- image = together_client.text_to_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="Stable Diffusion Image Generation",
42
- description="Generate images from text prompts using Stable Diffusion.",
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)