Update app.py

app.py

@@ -20,36 +20,9 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
 
 
 HEADER = """
-# TripoSR Demo
-<table bgcolor="#1E2432" cellspacing="0" cellpadding="0"  width="450">
-<tr style="height:50px;">
-<td style="text-align: center;">
-<a href="https://stability.ai">
-<img src="https://images.squarespace-cdn.com/content/v1/6213c340453c3f502425776e/6c9c4c25-5410-4547-bc26-dc621cdacb25/Stability+AI+logo.png" width="200" height="40" />
-</a>
-</td>
-<td style="text-align: center;">
-<a href="https://www.tripo3d.ai">
-<img src="https://tripo-public.cdn.bcebos.com/logo.png" width="40" height="40" />
-</a>
-</td>
-</tr>
-</table>
-<table bgcolor="#1E2432" cellspacing="0" cellpadding="0"  width="450">
-<tr style="height:30px;">
-<td style="text-align: center;">
-<a href="https://huggingface.co/stabilityai/TripoSR"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20Model_Card-Huggingface-orange" height="20"></a>
-</td>
-<td style="text-align: center;">
-<a href="https://github.com/VAST-AI-Research/TripoSR"><img src="https://postimage.me/images/2024/03/04/GitHub_Logo_White.png" width="100" height="20"></a>
-</td>
-<td style="text-align: center; color: white;">
-<a href="https://arxiv.org/abs/2403.02151"><img src="https://img.shields.io/badge/arXiv-2403.02151-b31b1b.svg" height="20"></a>
-</td>
-</tr>
-</table>
-
-**TripoSR** is a state-of-the-art open-source model for **fast** feedforward 3D reconstruction from a single image, developed in collaboration between [Tripo AI](https://www.tripo3d.ai/) and [Stability AI](https://stability.ai/).
+# Real3D Demo
+
+**Real3D** is a single-view large reconstruction model with self-training on real-world images.
 
 **Tips:**
 1. If you find the result is unsatisfied, please try to change the foreground ratio. It might improve the results.
@@ -64,9 +37,9 @@ else:
     device = "cpu"
 
 model = TSR.from_pretrained(
-    "stabilityai/TripoSR",
+    "hwjiang/Real3D",
     config_name="config.yaml",
-    weight_name="model.ckpt",
+    weight_name="model_both_trained_v1.ckpt",
 )
 model.renderer.set_chunk_size(131072)
 model.to(device)