Update app.py

app.py

@@ -1,16 +1,18 @@
 import gradio as gr
 import numpy as np
+import random
 import torch
 from diffusers import DiffusionPipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Текущая/последняя загруженная модель
+model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+
 if torch.cuda.is_available():
     torch_dtype = torch.float16
 else:
     torch_dtype = torch.float32
 
-# Изначально загружаем модель по умолчанию
+# Изначально загружаем модель по умолчанию (как в исходном коде)
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
@@ -30,17 +32,15 @@ def infer(
 ):
     global model_repo_id, pipe
     
-    # Проверяем, нужно ли менять модель
+    # Если пользователь ввёл новую модель, пробуем загрузить
     if model != model_repo_id:
         try:
-            # Пробуем загрузить новую модель
             new_pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype)
             new_pipe = new_pipe.to(device)
-            # Если успешно, то обновляем pipe и модель
             pipe = new_pipe
             model_repo_id = model
         except Exception as e:
-            raise gr.Error(f"Не удалось загрузить модель {model}. Ошибка: {str(e)}")
+            raise gr.Error(f"Не удалось загрузить модель '{model}'.\nОшибка: {e}")
 
     generator = torch.Generator(device=device).manual_seed(seed)
 
@@ -69,17 +69,17 @@ css = """
 }
 """
 
+# Убрали выпадающее меню, заменили на текстовое поле
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image App")
 
-        # Вместо выпадающего списка — текстовое поле для ввода модели
         model = gr.Textbox(
-            label="Model name or path",
+            label="Model (repo or local path)",
             value="stabilityai/sdxl-turbo",  # Значение по умолчанию
             interactive=True
         )
-        
+
         prompt = gr.Text(
             label="Prompt",
             show_label=False,
@@ -102,7 +102,7 @@ with gr.Blocks(css=css) as demo:
             step=1,
             value=42,
         )
-
+        
         guidance_scale = gr.Slider(
             label="Guidance scale",
             minimum=0.0,