Update app.py

app.py

@@ -1,23 +1,23 @@
 import streamlit as st
 from PIL import Image
 import torch
-from diffusers import DiffusionPipeline
-import io
+from diffusers import DiffusionPipeline  # تأكد من استخدام مكتبة متوافقة مع النموذج الجديد
 import numpy as np
+import io
 
 # إعداد الصفحة
-st.set_page_config(page_title="InstantMesh - 3D Mesh Generation", page_icon="🖼️")
+st.set_page_config(page_title="3D Mesh Generation", page_icon="🖼️")
 
 # عنوان الصفحة
-st.title("InstantMesh - 3D Mesh Generation from Image")
+st.title("3D Mesh Generation from Image")
 
 # تعليمات
-st.write("Upload an image to generate a 3D mesh. InstantMesh creates detailed 3D models within seconds.")
+st.write("Upload an image to generate a 3D mesh.")
 
 # تحميل النموذج
 @st.cache_resource
 def load_model():
-    model_id = "TencentARC/InstantMesh"
+    model_id = "your_new_model_id_here"  # استبدل بـ ID النموذج الجديد
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
     return pipe
 
@@ -36,14 +36,12 @@ if uploaded_image is not None:
         
         # توليد الشبكة ثلاثية الأبعاد
         with st.spinner("Generating 3D mesh..."):
-            mesh = pipe(image_tensor).images
+            mesh = pipe(image_tensor).images  # تأكد من أن هذا هو الاستخدام الصحيح للنموذج
 
         st.success("3D mesh generated successfully!")
 
         # عرض النتيجة
         st.write("### 3D Mesh Preview")
-        # حاوية لعرض الـ 3D Mesh
-        # ملاحظة: قد تحتاج إلى تكوين طريقة لعرض أو تحميل النموذج ثلاثي الأبعاد بشكل صحيح.
         st.image(mesh[0], caption="Generated 3D Mesh", use_column_width=True)
 
         # تنزيل النموذج
@@ -53,6 +51,5 @@ if uploaded_image is not None:
 
 # تقديم بعض المعلومات حول النموذج
 st.write("""
-### About InstantMesh:
-InstantMesh is a feed-forward framework for instant 3D mesh generation from a single image. It uses a combination of a multiview diffusion model and a sparse-view reconstruction model based on the LRM architecture to create diverse 3D assets quickly. The framework integrates a differentiable iso-surface extraction module to optimize the mesh representation, making it highly efficient and accurate.
-""")
+### About the Model:
+This model generates 3D meshes from single images