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() | |