Update app.py

app.py

@@ -5,8 +5,8 @@ import spaces
 from diffusers import DiffusionPipeline
 import torch
 
+# Подключение к устройству
 device = "cuda" if torch.cuda.is_available() else "cpu"
-
 model_repo_id = "stabilityai/stable-diffusion-3-medium-diffusers"
 
 if torch.cuda.is_available():
@@ -17,43 +17,35 @@ else:
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 512
+MAX_IMAGE_SIZE = 512  # Максимальный размер изображения
 
-def calculate_duration(width, height, num_inference_steps):
-    # Примерная оценка времени выполнения
-    base_time = 10  # Минимальное время для 512x512, 10 шагов
-    time_factor = (width * height) / (512 * 512)  # Увеличение времени от размера
-    step_factor = num_inference_steps / 10  # Увеличение времени от числа шагов
-    return int(base_time * time_factor * step_factor)
+# Примеры для интерфейса
 examples = [
     "A border collie lying in some Fall leaves as the forest trees change colors",
+    "A cyberpunk cityscape with neon lights and flying cars at night",
+    "A fantasy castle surrounded by mountains under a pink and purple sunset",
 ]
-@spaces.GPU(duration=30)  # Устанавливаем 30 секунд
+
+# Функция генерации изображений
+@spaces.GPU(duration=30)  # Устанавливаем время выполнения задачи
 def infer(
     prompt,
     width=512,
     height=512,
-    num_inference_steps=20,  # Оптимальные шаги
-    guidance_scale=7.5,
+    num_inference_steps=20,  # Количество итераций
+    guidance_scale=7.5,  # Вес промпта
 ):
-    # Оценка необходимого времени
-    duration = calculate_duration(width, height, num_inference_steps)
-    print(f"Запрашиваемая квота: {duration} секунд")
-    # Настраиваем параметры в pipe
+    # Генерация изображения
     image = pipe(
         prompt=prompt,
         num_inference_steps=num_inference_steps,
+        guidance_scale=guidance_scale,
         width=width,
         height=height,
-        guidance_scale=guidance_scale,
     ).images[0]
     return image
 
-
-  
-
-# Define the custom theme with input styles
+# Класс для пользовательской темы интерфейса
 class CustomTheme(gr.themes.Base):
     def __init__(self):
         super().__init__()
@@ -64,12 +56,12 @@ class CustomTheme(gr.themes.Base):
         self.text_color_primary = "#AEB3B8"
         self.text_color_secondary = "#AEB3B8"
         self.text_color_tertiary = "#AEB3B8"
-        self.input_background_fill = "#17181B"  # Set input background color
-        self.input_text_color = "#AEB3B8"       # Set input text color
+        self.input_background_fill = "#17181B"  # Цвет фона для ввода текста
+        self.input_text_color = "#AEB3B8"       # Цвет текста для ввода
 
-# Custom CSS to hide the footer, set fonts, and adjust font weights
+# CSS для интерфейса
 css = """
-/* Hide the footer */
+/* Скрываем нижний колонтитул */
 footer {
     visibility: hidden;
     height: 0;
@@ -78,26 +70,21 @@ footer {
     overflow: hidden;
 }
 
-/* Import Google Fonts */
+/* Импортируем шрифты */
 @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;700&family=Montserrat:wght@500;700&display=swap');
 
-/* Apply fonts to different elements */
+/* Применяем шрифты */
 body, input, button, textarea, select, .gr-button {
     font-family: 'Roboto', sans-serif;
 }
 
-/* Make button text normal weight */
-.generate-button, .generate-button .gr-button {
-    font-weight: normal !important;
-}
-
-/* Ensure headings use Montserrat */
+/* Настройки заголовков */
 h1, h2, h3, h4, h5, h6 {
     font-family: 'Montserrat', sans-serif;
     font-weight: 700;
 }
 
-/* Additional styling for sliders and checkboxes if needed */
+/* Дополнительный стиль для слайдеров и чекбоксов */
 input[type="range"]::-webkit-slider-thumb {
     background: #5271FF;
 }
@@ -111,15 +98,16 @@ input[type="checkbox"]:checked {
     background-color: #5271FF;
 }
 
-/* Make Prompt text bold */
+/* Выделяем текст для Prompt */
 .prompt-text {
     font-weight: bold;
 }
 """
 
+# Создание интерфейса Gradio
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        # Make "Prompt" bold using Markdown syntax and assign a class
+        # Заголовок для ввода промпта
         gr.Markdown("**Prompt**", elem_classes="prompt-text")
 
         with gr.Row():
@@ -140,21 +128,22 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
 
         result = gr.Image(label="Result", show_label=False)
 
-       
-    prompt.submit(
-        fn=infer,
-        inputs=[
-            prompt,
-            negative_prompt,
-            seed,
-            randomize_seed,
-            width,
-            height,
-            guidance_scale,
-            num_inference_steps,
-        ],
-        outputs=[result, seed],
-    )
+        # Примеры для интерфейса
+        gr.Examples(
+            examples=examples,
+            inputs=[prompt],
+            outputs=[result],
+            fn=infer,
+            cache_examples=True,
+            cache_mode="lazy"
+        )
+
+        # Подключение кнопки к функции
+        run_button.click(
+            fn=infer,
+            inputs=[prompt],
+            outputs=[result],
+        )
 
 if __name__ == "__main__":
     demo.launch(