Update app.py

app.py

@@ -1,5 +1,20 @@
-from transformers import AutoImageProcessor, AutoModelForDepthEstimation, DepthAnythingForDepthEstimation
+import gradio as gr
+from transformers import DepthAnythingForDepthEstimation
 
+# 加载模型
 model = DepthAnythingForDepthEstimation.from_pretrained("depth-anything/Depth-Anything-V2-Small-hf")
- 
-print(model.backbone)
+backbone = model.backbone
+config = backbone.config
+
+# 定义获取主干信息的函数
+def get_backbone_info():
+ info = f"主干: {backbone.__class__.__name__}\n"
+ info += f"隐藏层大小: {config.hidden_size}\n"
+ info += f"层数: {config.num_hidden_layers}\n"
+ info += f"注意力头数: {config.num_attention_heads}\n"
+ info += f"补丁大小: {config.patch_size}\n"
+ return info
+
+# 创建 Gradio 接口
+demo = gr.Interface(fn=get_backbone_info, inputs=[], outputs="text")
+demo.launch()