Ryukijano commited on
Commit
f2e0064
·
verified ·
1 Parent(s): 53f1f2c

Update app.py

Browse files

Added the import statements that were deleted by accident before.

Files changed (1) hide show
  1. app.py +16 -0
app.py CHANGED
@@ -1,3 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  def main():
2
  print("[INFO] Starting main function...")
3
  if torch.cuda.is_available():
 
1
+ import sys
2
+ import spaces
3
+ sys.path.append("flash3d") # Add the flash3d directory to the system path for importing local modules
4
+
5
+ from omegaconf import OmegaConf
6
+ import gradio as gr
7
+ import torch
8
+ import torchvision.transforms as TT
9
+ import torchvision.transforms.functional as TTF
10
+ from huggingface_hub import hf_hub_download
11
+ import numpy as np
12
+
13
+ from networks.gaussian_predictor import GaussianPredictor
14
+ from util.vis3d import save_ply
15
+
16
+
17
  def main():
18
  print("[INFO] Starting main function...")
19
  if torch.cuda.is_available():