Update app.py

feat: Add GPU support and model configuration download

- Added GPU support detection and device setting in the main function.
- Implemented downloading of model configuration and weights from Hugging Face Hub.
- Included necessary imports for Gradio, Torch, and other dependencies.
- Added system path for importing local modules from the flash3d directory.
- Improved logging to provide detailed information about the process.

app.py

@@ -92,6 +92,11 @@ def main():
         print("[INFO] Passing image through the model...")
         outputs = model(inputs)
 
+        #Ensure the tensor dimensions are compatible
+        gauss_means = outputs[('gauss_means',0, 0)]
+        if gauss_means.shape[0] % num_gauss != 0:
+            raise ValueError(f"Shape mismatch: cannot divide axis of length {gauss_means.shape[0]} into chunks of {num_gauss}")
+
         # Export the reconstruction to a PLY file
         print(f"[INFO] Saving output to {ply_out_path}...")
         save_ply(outputs, ply_out_path, num_gauss=num_gauss)