Update app.py

app.py

@@ -21,14 +21,14 @@ def load_model():
             'openbmb/MiniCPM-o-2_6',
             trust_remote_code=True,
             attn_implementation='sdpa',
-            torch_dtype=torch.float32,  # CPU推理使用float32
+            torch_dtype=torch.bfloat16,
             init_vision=True,
             init_audio=False,
             init_tts=False
         )
-        model = model.eval()  # 移除.cuda()，使用CPU
+        model = model.eval().cuda()
         tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-o-2_6', trust_remote_code=True)
-        print("模型加载完成 (CPU模式)")
+        print("模型加载完成")
     return model, tokenizer
 
 def clean_markdown_output(text):
@@ -292,7 +292,7 @@ def create_interface():
         ### 🔧 技术支持
         - 模型: MiniCPM-o-2.6
         - 框架: Gradio + Transformers
-        - 推理: CPU模式
+        - GPU: CUDA加速推理
         """)
     
     return interface