Hugging Face's logo

Spaces:
greykingreys
/
generator_images
Sleeping

App Files Community
greykingreys commited on
Commit
96d9d8b
·
verified ·
1 Parent(s): 898d2f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -10,7 +10,6 @@ img_generator1.to("cuda" if torch.cuda.is_available() else "cpu")
10
  img_generator2 = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
11
  img_generator2.to("cuda" if torch.cuda.is_available() else "cpu")
12
 
13
-
14
  # Fonction de génération d’image à partir du texte
15
  def text_to_image(prompt, model_name):
16
  if model_name == 'kandinsky-2-2-decoder':
 
10
  img_generator2 = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
11
  img_generator2.to("cuda" if torch.cuda.is_available() else "cpu")
12
 
 
13
  # Fonction de génération d’image à partir du texte
14
  def text_to_image(prompt, model_name):
15
  if model_name == 'kandinsky-2-2-decoder':