Update app.py

app.py

@@ -14,8 +14,8 @@ subprocess.run(
 )
 
 pipeline = DiffusionPipeline.from_pretrained(
-    "ahmetyaylalioglu/multiview-diffusion-synthesis",
-    custom_pipeline="ahmetyaylalioglu/multiview-diffusion-synthesis",
+    "dylanebert/LGM-full",
+    custom_pipeline="dylanebert/LGM-full",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")
@@ -24,18 +24,12 @@ pipeline = DiffusionPipeline.from_pretrained(
 @spaces.GPU
 def run(image):
     input_image = np.array(image, dtype=np.float32) / 255.0
-    try:
-        splat = pipeline("", input_image, guidance_scale=5, num_inference_steps=30, elevation=0)
-        splat_file = "/tmp/output.ply"
-        # Assuming `save_ply` is a function you defined to handle .ply files.
-        save_ply(splat, splat_file)
-        return splat_file
-    except AttributeError as e:
-        return f"Failed to save file due to an attribute error: {str(e)}"
-    except Exception as e:
-        return f"An error occurred: {str(e)}"
-
-    
+    splat = pipeline(
+        "", input_image, guidance_scale=5, num_inference_steps=30, elevation=0
+    )
+    splat_file = "/tmp/output.ply"
+    pipeline.save_ply(splat, splat_file)
+    return splat_file
 
 
 demo = gr.Interface(
@@ -50,4 +44,4 @@ demo = gr.Interface(
     cache_examples=True,
     allow_duplication=True,
 )
-demo.queue().launch()
+demo.queue().launch()