Spaces:
Paused
Paused
update reqs
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -30,7 +30,7 @@ prompt = f"A photo of {unique_id} {class_label} in a bucket"
|
|
| 30 |
def infer(prompt):
|
| 31 |
generated_images = pipeline(prompt, guidance_scale=10, num_inference_steps=50, num_images_per_prompt=2)
|
| 32 |
|
| 33 |
-
return generated_images
|
| 34 |
|
| 35 |
output = gr.Gallery(label="Outputs").style(grid=(1,2))
|
| 36 |
|
|
|
|
| 30 |
def infer(prompt):
|
| 31 |
generated_images = pipeline(prompt, guidance_scale=10, num_inference_steps=50, num_images_per_prompt=2)
|
| 32 |
|
| 33 |
+
return generated_images.images
|
| 34 |
|
| 35 |
output = gr.Gallery(label="Outputs").style(grid=(1,2))
|
| 36 |
|
requirements.txt
CHANGED
|
@@ -6,3 +6,4 @@ huggingface-hub
|
|
| 6 |
diffusers
|
| 7 |
transformers
|
| 8 |
pycocotools
|
|
|
|
|
|
| 6 |
diffusers
|
| 7 |
transformers
|
| 8 |
pycocotools
|
| 9 |
+
accelerate
|