Update app.py

app.py

@@ -65,6 +65,15 @@ class SimpleModelManager:
     
     def cleanup(self):
         pass
+    
+    def move_models_to_cpu(self):
+        pass
+    
+    def move_models_to_gpu(self):
+        pass
+    
+    def force_reload_all_models(self):
+        pass
 
 # 全局状态管理
 class AppState:
@@ -551,8 +560,17 @@ def create_gradio_interface():
         with gr.Row():
             with gr.Column():
                 gr.Markdown("### 🔧 系统控制")
-                cleanup_btn = gr.Button("🧹 清理内存", variant="secondary")
-                gr.Markdown("> **提示**: 如果遇到内存问题，请点击清理内存按钮")
+                cleanup_btn = gr.Button("🧹 清理显存缓存", variant="secondary")
+                cpu_btn = gr.Button("💾 模型移至CPU", variant="secondary") 
+                gpu_btn = gr.Button("🚀 模型移至GPU", variant="secondary")
+                reload_btn = gr.Button("🔄 重新加载模型", variant="primary")
+                gr.Markdown("""
+                > **使用说明**: 
+                > - **清理显存缓存**: 只清理缓存，不影响模型
+                > - **模型移至CPU**: 释放GPU显存，但推理会变慢
+                > - **模型移至GPU**: 将模型移回GPU，恢复正常速度
+                > - **重新加载模型**: 如果模型出现问题，强制重新加载
+                """)
             
         # 事件绑定 - 完整的流水线连接
         analyze_btn.click(
@@ -579,6 +597,24 @@ def create_gradio_interface():
             outputs=[]
         )
         
+        cpu_btn.click(
+            fn=model_manager.move_models_to_cpu,
+            inputs=[],
+            outputs=[]
+        )
+        
+        gpu_btn.click(
+            fn=model_manager.move_models_to_gpu,
+            inputs=[],
+            outputs=[]
+        )
+        
+        reload_btn.click(
+            fn=model_manager.force_reload_all_models,
+            inputs=[],
+            outputs=[]
+        )
+        
         # 添加使用说明
         with gr.Row():
             gr.Markdown("""
@@ -592,6 +628,7 @@ def create_gradio_interface():
             - 每一步都会调用相应的AI模型，请耐心等待生成完成
             - 使用ControlNet模型生成高质量的3D试穿效果
             - 所有设计都基于原始图片的分析结果
+            - 如果遇到模型问题，可以使用下方的系统控制按钮
             """)
     
     return demo