Update app.py

app.py

@@ -3,7 +3,6 @@ import numpy as np
 import random
 from diffusers import DiffusionPipeline
 import torch
-
 import time
 import psutil
 
@@ -42,7 +41,7 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
     if not prompt:
         raise gr.Error("Будь ласка, введіть опис для зображення.")
 
-    torch.set_num_threads(MAX_THREADS)  # Set the maximum number of threads
+    torch.set_num_threads(MAX_THREADS) 
 
     pipe = load_pipeline(MODEL_OPTIONS[model_choice])
 
@@ -70,11 +69,30 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
 
     return images, f"Час генерації: {generation_time:.2f} секунд"
 
-# ... (Gradio interface remains the same)
-
-    generation_time = end_time - start_time
-
-    return images, f"Час генерації: {generation_time:.2f} секунд"
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=5):
+            prompt = gr.Textbox(label="Опис зображення")
+            negative_prompt = gr.Textbox(label="Негативний опис", value="")
+        with gr.Column(scale=1):
+            model_choice = gr.Radio(
+                choices=list(MODEL_OPTIONS.keys()),
+                label="Якість моделі",
+                value=list(MODEL_OPTIONS.keys())[0],
+            )
+    with gr.Row():
+        seed = gr.Slider(label="Seed", minimum=0, maximum=1000000, step=1, value=42)
+        randomize_seed = gr.Checkbox(label="Випадковий Seed", value=True)
+
+    with gr.Row():
+        width = gr.Slider(label="Ширина", minimum=256, maximum=1024, step=64, value=DEFAULT_IMAGE_SIZE)
+        height = gr.Slider(label="Висота", minimum=256, maximum=1024, step=64, value=DEFAULT_IMAGE_SIZE)
+    with gr.Row():
+        guidance_scale = gr.Slider(label="Guidance Scale", minimum=0, maximum=20, step=0.5, value=7.5)
+        num_inference_steps = gr.Slider(label="Кроки інференсу", minimum=10, maximum=100, step=5, value=50)
+
+    with gr.Row():
+        num_images = gr.Slider(label="Кількість зображень", minimum=1, maximum=4, step=1, value=1)
 
     run_button = gr.Button("Згенерувати")
     gallery = gr.Gallery(label="Згенеровані зображення")
@@ -83,5 +101,8 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
     run_button.click(
         fn=generate_image,
         inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, num_images, model_choice],
-        outputs=[gallery, status_text],  # Output both the gallery and status text
+        outputs=[gallery, status_text],
     )
+        
+
+demo.launch(share=True)