midas / gradio_app.py
adpro's picture
Create gradio_app.py
b1959a5 verified
raw
history blame
644 Bytes
import gradio as gr
from PIL import Image
def get_depth_map():
"""Trả về ảnh Depth Map mới nhất từ FastAPI"""
try:
return Image.open("depth_map.png") # 🟢 Lấy ảnh đã lưu từ FastAPI
except FileNotFoundError:
return "Chưa có ảnh Depth Map nào được xử lý!"
# 🟢 Tạo UI trên Hugging Face Spaces với Gradio
gr.Interface(
fn=get_depth_map,
inputs=[],
outputs=gr.Image(type="pil"),
title="🔍 Depth Map Viewer",
description="Hiển thị ảnh Depth Map mới nhất từ FastAPI",
live=True # 🎯 **Tự động cập nhật ảnh mới**
).launch()