Update app.py

app.py

@@ -5,12 +5,13 @@ from PIL import Image
 import re
 import os
 import requests
+import uuid
+import base64 
 
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt")
 word_list = word_list_dataset["train"]['text']
-print(word_list)
 
 is_gpu_busy = False
 def infer(prompt, negative, scale):
@@ -27,11 +28,12 @@ def infer(prompt, negative, scale):
     payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
     images_request = requests.post(url, json = payload)
     for image in images_request.json()["images"]:
-        print("Loading inside")
-        image_b64 = (f"data:image/jpeg;base64,{image}")
-        images.append(image_b64)
-    print(images)
-    return images
+        file_path = f"{uuid.uuid4()}.jpg"
+        with open(file_path, "wb") as f:
+            f.write(base64.b64decode(image))
+        images.append(file_path)
+    
+    return images 
     
     
 css = """
@@ -253,7 +255,7 @@ with block:
     )
     with gr.Group():
         with gr.Box():
-            with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
+            with gr.Row(elem_id="prompt-container"):
                 with gr.Column():
                     text = gr.Textbox(
                         label="Enter your prompt",
@@ -261,10 +263,6 @@ with block:
                         max_lines=1,
                         placeholder="Enter your prompt",
                         elem_id="prompt-text-input",
-                    ).style(
-                        border=(True, False, True, True),
-                        rounded=(True, False, False, True),
-                        container=False,
                     )
                     negative = gr.Textbox(
                         label="Enter your negative prompt",
@@ -272,20 +270,12 @@ with block:
                         max_lines=1,
                         placeholder="Enter a negative prompt",
                         elem_id="negative-prompt-text-input",
-                    ).style(
-                        border=(True, False, True, True),
-                        rounded=(True, False, False, True),
-                        container=False,
                     )
-                btn = gr.Button("Generate image").style(
-                    margin=False,
-                    rounded=(False, True, True, False),
-                    full_width=False,
-                )
+                btn = gr.Button("Generate image")
 
         gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+        )
 
         with gr.Group(elem_id="container-advanced-btns"):
             #advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")