Spaces:
Runtime error
Runtime error
go back to random file names
Browse files
app.py
CHANGED
|
@@ -4,8 +4,12 @@ import sys
|
|
| 4 |
import zipfile
|
| 5 |
import requests
|
| 6 |
import tempfile
|
| 7 |
-
import json
|
| 8 |
from io import BytesIO
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
# gradio / hf stuff
|
| 11 |
import gradio as gr
|
|
@@ -69,9 +73,12 @@ def download_image(url):
|
|
| 69 |
def zip_images(image_paths_and_labels):
|
| 70 |
zip_file_path = tempfile.NamedTemporaryFile(delete=False, suffix='.zip').name
|
| 71 |
with zipfile.ZipFile(zip_file_path, 'w') as zipf:
|
| 72 |
-
for image_url,
|
| 73 |
image_content = download_image(image_url)
|
| 74 |
-
|
|
|
|
|
|
|
|
|
|
| 75 |
return zip_file_path
|
| 76 |
|
| 77 |
|
|
@@ -136,7 +143,7 @@ with gr.Blocks() as demo:
|
|
| 136 |
text = gr.Textbox(label="What do you want to create?",
|
| 137 |
placeholder="Enter your text and then click on the \"Image Generate\" button")
|
| 138 |
|
| 139 |
-
model = gr.Dropdown(choices=["dall-e-2", "dall-e-3"], label="Model", value="dall-e-
|
| 140 |
btn = gr.Button("Generate Images")
|
| 141 |
output_images = gr.Gallery(label="Image Outputs", show_label=True, columns=[3], rows=[1], object_fit="contain",
|
| 142 |
height="auto", allow_preview=False)
|
|
|
|
| 4 |
import zipfile
|
| 5 |
import requests
|
| 6 |
import tempfile
|
|
|
|
| 7 |
from io import BytesIO
|
| 8 |
+
import random
|
| 9 |
+
import string
|
| 10 |
+
|
| 11 |
+
#image generation stuff
|
| 12 |
+
from PIL import Image
|
| 13 |
|
| 14 |
# gradio / hf stuff
|
| 15 |
import gradio as gr
|
|
|
|
| 73 |
def zip_images(image_paths_and_labels):
|
| 74 |
zip_file_path = tempfile.NamedTemporaryFile(delete=False, suffix='.zip').name
|
| 75 |
with zipfile.ZipFile(zip_file_path, 'w') as zipf:
|
| 76 |
+
for image_url, _ in image_paths_and_labels:
|
| 77 |
image_content = download_image(image_url)
|
| 78 |
+
# Generate a random filename for the image
|
| 79 |
+
random_filename = ''.join(random.choices(string.ascii_letters + string.digits, k=10)) + ".png"
|
| 80 |
+
# Write the image content to the zip file with the random filename
|
| 81 |
+
zipf.writestr(random_filename, image_content)
|
| 82 |
return zip_file_path
|
| 83 |
|
| 84 |
|
|
|
|
| 143 |
text = gr.Textbox(label="What do you want to create?",
|
| 144 |
placeholder="Enter your text and then click on the \"Image Generate\" button")
|
| 145 |
|
| 146 |
+
model = gr.Dropdown(choices=["dall-e-2", "dall-e-3"], label="Model", value="dall-e-2")
|
| 147 |
btn = gr.Button("Generate Images")
|
| 148 |
output_images = gr.Gallery(label="Image Outputs", show_label=True, columns=[3], rows=[1], object_fit="contain",
|
| 149 |
height="auto", allow_preview=False)
|