Upload app.py

app.py

@@ -8,8 +8,6 @@ import torch
 model = test_1.load_trained_model()
 
 
-# 判断是否在 Hugging Face 环境中
-is_huggingface = 'HF_HOME' in os.environ
 
 def convert_to_28x28_grid(canvas_image):
     #将750x750画布转换为28x28数值矩阵
@@ -69,23 +67,13 @@ with gr.Blocks() as demo:
     gr.Markdown("""# digit-recognition""")
 
     with gr.Row():
-        if is_huggingface:
-            # 在 Hugging Face 上，使用 height 和 width，而不是 shape 参数
-            input_image = gr.Sketchpad(
-                height=750,
-                width=750,
-                brush_color="black",
-                image_mode="L",  # 确保输出是单通道
-                type="pil"
-            )
-        else:
-            # 本地使用 shape 参数
-            input_image = gr.Sketchpad(
-                shape=(750, 750),
-                brush_color="black",
-                image_mode="L",  # 确保输出是单通道
-                type="pil"
-            )
+        # 本地使用 shape 参数
+        input_image = gr.Sketchpad(
+            shape=(750, 750),
+            brush_color="black",
+            image_mode="L",  # 确保输出是单通道
+            type="pil"
+        )
 
 
     with gr.Row():