Update app.py

app.py

@@ -1,22 +1,18 @@
-# import gradio as gr
-
-# gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA").launch()
-
 import gradio as gr
 import random
 
-# 首先加载模型
+# 加载 Hugging Face 模型，返回 Gradio Blocks 对象
 model = gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA")
 
-# 定义生成函数，并接收种子参数
+# 定义生成函数，使用 predict() 方法并传入 seed 参数
 def generate_image(prompt, seed=None):
     if seed is None:
-        seed = random.randint(0, 1e6)  # 随机生成种子
-    # 设置生成参数，包括种子
-    result = model(prompt=prompt, seed=seed)
+        seed = random.randint(0, 1e6)  # 如果用户未指定种子，则生成随机数
+    # 将 prompt 和 seed 作为列表传入 predict 方法
+    result = model.predict(prompt, seed)
     return result
 
-# 使用 Gradio Blocks 创建自定义界面
+# 使用 Gradio Blocks 创建界面
 with gr.Blocks() as demo:
     gr.Markdown("## 输入描述和种子以生成图像")
     prompt = gr.Textbox(label="描述")