Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import gradio as gr
 import numpy as np
 import random
@@ -7,7 +9,7 @@ from huggingface_hub import InferenceClient
 # 初始化 InferenceClient
 client = InferenceClient(
     provider="fal-ai",
-    api_key=os.environ["HF_TOKEN"],
+    api_key=os.environ.get("HF_TOKEN", ""),
 )
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -24,25 +26,23 @@ def infer(
     num_inference_steps,
     progress=gr.Progress(track_tqdm=True),
 ):
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    # 拼接 prompt 和 negative_prompt
+    if not prompt or prompt.strip() == "":
+        return None, seed
+
+    # 只传递 prompt，避免参数过多导致报错
     full_prompt = prompt
     if negative_prompt:
         full_prompt += f". Negative prompt: {negative_prompt}"
 
-    # 发送推理请求
-    image = client.text_to_image(
-        full_prompt,
-        model="black-forest-labs/FLUX.1-Krea-dev",
-        # 下面参数部分模型可能不支持，可根据实际情况调整
-        # width=width,
-        # height=height,
-        # guidance_scale=guidance_scale,
-        # num_inference_steps=num_inference_steps,
-        seed=seed,
-    )
-    return image, seed
+    try:
+        image = client.text_to_image(
+            full_prompt,
+            model="black-forest-labs/FLUX.1-Krea-dev",
+        )
+        return image, seed
+    except Exception as e:
+        # 返回错误信息到前端
+        return f"生成图片出错: {str(e)}", seed
 
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
@@ -72,7 +72,7 @@ with gr.Blocks(css=css) as demo:
 
             run_button = gr.Button("Run", scale=0, variant="primary")
 
-        result = gr.Image(label="Result", show_label=False)
+        result = gr.Image(label="Result", show_label=False, type="pil")
 
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Text(