Update app.py

app.py

@@ -1,29 +1,50 @@
+import os
+os.environ.setdefault("SYSTEM", "spaces")
+
+# ===== Monkey Patch: 修复 gradio_client 的 bool-schema Bug =====
+import gradio_client.utils as _gcu
+
+if hasattr(_gcu, "_json_schema_to_python_type"):
+    _orig = _gcu._json_schema_to_python_type
+    def _safe(schema, defs=None):
+        if isinstance(schema, bool):
+            return "Any"
+        return _orig(schema, defs)
+    _gcu._json_schema_to_python_type = _safe
+
+if hasattr(_gcu, "get_type"):
+    _orig_get_type = _gcu.get_type
+    def _safe_get_type(schema):
+        if not isinstance(schema, dict):
+            return "Any"
+        return _orig_get_type(schema)
+    _gcu.get_type = _safe_get_type
+# ===== Monkey Patch 结束 =====
+
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from PIL import Image
 
 # ============================================================
-# 模型初始化（纯 CPU 环境，直接加载到内存）
+# 模型初始化
 # ============================================================
 model_id = "vikhyatk/moondream2"
 revision = "2024-08-26"
 
 print("正在加载 Moondream2 模型...")
 tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
-
-# 注意：CPU 环境下我们使用 float32 精度，兼容性最好
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     trust_remote_code=True,
     revision=revision,
-    torch_dtype=torch.float32, 
-    low_cpu_mem_usage=True # 降低 CPU 内存占用
+    torch_dtype=torch.float32,
+    low_cpu_mem_usage=True,
 )
 print("模型加载完成。")
 
 # ============================================================
-# 推理函数（纯 CPU 推理）
+# 推理函数
 # ============================================================
 def answer_question(image: Image.Image, question: str):
     if image is None:
@@ -31,10 +52,8 @@ def answer_question(image: Image.Image, question: str):
     if not question or not question.strip():
         return "⚠️ 请输入你的问题。"
 
-    # Moondream2 的标准问答流程
     enc_image = model.encode_image(image)
     answer = model.answer_question(enc_image, question, tokenizer)
-
     return answer
 
 # ============================================================
@@ -52,9 +71,13 @@ demo = gr.Interface(
     ],
     outputs=gr.Textbox(label="🤖 AI 回答", lines=8),
     title="📚 作业识图 AI (CPU 版)",
-    description="上传作业照片，输入问题，AI 会尝试识别并给出答案。当前运行在 CPU basic 上，推理速度可能稍慢，请耐心等待。",
+    description="上传作业照片，输入问题，AI 会尝试识别并给出答案。",
     theme=gr.themes.Soft(),
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True,
+    )