anusha-bhambore commited on
Commit
63b7b9a
·
verified ·
1 Parent(s): 449842d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,9 +5,9 @@ import gradio as gr
5
  device = "cuda" if torch.cuda.is_available() else "cpu"
6
  pipeline = DiffusionPipeline.from_pretrained("anusha-bhambore/live-eventful")
7
  pipeline = pipeline.to(device)
8
- def generate_image_interface(prompt, negative_prompt, gender, age, num_inference_steps=50, weight=640):
9
  params = {'prompt': prompt, 'num_inference_steps': num_inference_steps, 'num_images_per_prompt':2, 'height':int(1.2*weight),
10
- 'weight': weight, 'negative_prompt': negative_prompt,'gender': gender, 'age': age }
11
 
12
  img = pipeline(**params).images
13
  return img[0],img[1]
 
5
  device = "cuda" if torch.cuda.is_available() else "cpu"
6
  pipeline = DiffusionPipeline.from_pretrained("anusha-bhambore/live-eventful")
7
  pipeline = pipeline.to(device)
8
+ def generate_image_interface(prompt, negative_prompt, num_inference_steps=50, weight=640):
9
  params = {'prompt': prompt, 'num_inference_steps': num_inference_steps, 'num_images_per_prompt':2, 'height':int(1.2*weight),
10
+ 'weight': weight, 'negative_prompt': negative_prompt}
11
 
12
  img = pipeline(**params).images
13
  return img[0],img[1]