Update app.py

app.py

@@ -1,4 +1,8 @@
-# final one
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
 import torch
 import gradio as gr
 import os
@@ -18,8 +22,8 @@ from gradio_litmodel3d import LitModel3D
 # we load the pre-trained model from HF
 class LRMGeneratorWrapper:
     def __init__(self):
-        self.config = AutoConfig.from_pretrained("facebook/vfusion3d", trust_remote_code=True)
-        self.model = AutoModel.from_pretrained("facebook/vfusion3d", trust_remote_code=True)
+        self.config = AutoConfig.from_pretrained("jadechoghari/vfusion3d", trust_remote_code=True)
+        self.model = AutoModel.from_pretrained("jadechoghari/vfusion3d", trust_remote_code=True)
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.model.to(self.device)
         self.model.eval()
@@ -274,4 +278,4 @@ with gr.Blocks() as demo:
     generate_mesh_button.click(step_1_generate_obj, inputs=img_input, outputs=[obj_file_output, model_output])
     generate_video_button.click(step_2_generate_video, inputs=img_input, outputs=video_file_output)
 
-demo.launch(share=False)
+demo.launch(share=False)