Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 from model_pipelines import load_pipelines, generate_all
 from grace_eval import compute_sample_scores, plot_radar
+import torch
+
+# 强制使用CPU
+torch.backends.cuda.is_available = lambda: False
 
 pipes = load_pipelines()
 
@@ -14,14 +18,16 @@ def show_leaderboard(prompt):
     return "radar.png"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 图像生成模型对比实验")
+    gr.Markdown("# 图像生成模型对比实验 (CPU模式)")
 
     with gr.Tab("Arena"):
-        prompt = gr.Textbox(label="请输入生成文本")
-        btn = gr.Button("生成图像")
-        out1 = gr.Image(label="StableDiffusion v1.5")
-        out2 = gr.Image(label="Openjourney v4")
-        out3 = gr.Image(label="LDM 256")
+        with gr.Row():
+            prompt = gr.Textbox(label="请输入生成文本", scale=4)
+            btn = gr.Button("生成图像", scale=1)
+        with gr.Row():
+            out1 = gr.Image(label="StableDiffusion v1.5")
+            out2 = gr.Image(label="Openjourney v4")
+            out3 = gr.Image(label="LDM 256")
         btn.click(compare, inputs=prompt, outputs=[out1, out2, out3])
 
     with gr.Tab("Leaderboard"):