Update app.py

app.py

@@ -17,23 +17,23 @@ else:
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
-MAX_IMAGE_SIZE = 512  # Максимальный размер изображения
+MAX_IMAGE_SIZE = 512  
+
 
-# Примеры для интерфейса
 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)  # Устанавливаем время выполнения задачи
+
+@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,  
 ):
     # Генерация изображения
     image = pipe(
@@ -45,21 +45,20 @@ def infer(
     ).images[0]
     return image
 
-# Класс для пользовательской темы интерфейса
+
 class CustomTheme(gr.themes.Base):
     def __init__(self):
         super().__init__()
         self.primary_hue = "#5271FF"
-        self.background_fill_primary = "#17181B"
-        self.background_fill_secondary = "#17181B"
-        self.background_fill_tertiary = "#17181B"
+        self.background_fill_primary = "#AEB3B8"
+        self.background_fill_secondary = "#AEB3B8"
+        self.background_fill_tertiary = "#AEB3B8"
         self.text_color_primary = "#AEB3B8"
         self.text_color_secondary = "#AEB3B8"
         self.text_color_tertiary = "#AEB3B8"
-        self.input_background_fill = "#AEB3B8"  # Цвет фона для ввода текста
-        self.input_text_color = "#AEB3B8"       # Цвет текста для ввода
+        self.input_background_fill = "#AEB3B8"  
+        self.input_text_color = "#AEB3B8"       
 
-# CSS для интерфейса
 css = """
 /* Скрываем нижний колонтитул */
 footer {
@@ -70,20 +69,19 @@ footer {
     overflow: hidden;
 }
 
-/* Импортируем шрифты */
-@import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;700&family=Montserrat:wght@500;700&display=swap');
 
+@import url('https://fonts.googleapis.com/css2?family=Poppins:wght@400;500;700&display=swap');
 /* Применяем шрифты */
 body, input, button, textarea, select, .gr-button {
-    font-family: 'Roboto', sans-serif;
+    font-family: 'Poppins', sans-serif;
 }
-
 /* Настройки заголовков */
 h1, h2, h3, h4, h5, h6 {
-    font-family: 'Montserrat', sans-serif;
+    font-family: 'Poppins', sans-serif;
     font-weight: 700;
 }
 
+
 /* Дополнительный стиль для слайдеров и чекбоксов */
 input[type="range"]::-webkit-slider-thumb {
     background: #5271FF;
@@ -104,10 +102,10 @@ input[type="checkbox"]:checked {
 }
 """
 
-# Создание интерфейса Gradio
+
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        # Заголовок для ввода промпта
+
         gr.Markdown("**Prompt**", elem_classes="prompt-text")
 
         with gr.Row():
@@ -128,7 +126,7 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
 
         result = gr.Image(label="Result", show_label=False)
 
-        # Примеры для интерфейса
+
         gr.Examples(
             examples=examples,
             inputs=[prompt],
@@ -138,7 +136,7 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
             cache_mode="lazy"
         )
 
-        # Подключение кнопки к функции
+
         run_button.click(
             fn=infer,
             inputs=[prompt],