FrozenBurning commited on
Commit
ba74ae4
·
1 Parent(s): 1299110

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -9,6 +9,7 @@ from tqdm import tqdm
9
 
10
  os.system("git clone https://github.com/FrozenBurning/SceneDreamer.git")
11
  sys.path.append("SceneDreamer")
 
12
  pretrained_model = dict(file_url='https://drive.google.com/uc?id=1IFu1vNrgF1EaRqPizyEgN_5Vt7Fyg0Mj',
13
  alt_url='', file_size=330571863,
14
  file_path='./scenedreamer_released.pt',)
@@ -90,14 +91,9 @@ import torch
90
  import argparse
91
  from imaginaire.config import Config
92
  from imaginaire.utils.cudnn import init_cudnn
93
- from imaginaire.utils.dataset import get_test_dataloader
94
- from imaginaire.utils.distributed import init_dist
95
- from imaginaire.utils.gpu_affinity import set_affinity
96
  from imaginaire.utils.io import get_checkpoint as get_checkpoint
97
- from imaginaire.utils.logging import init_logging
98
  from imaginaire.utils.trainer import \
99
  (get_model_optimizer_and_scheduler, set_random_seed)
100
- import imaginaire.config
101
  import gradio as gr
102
  from PIL import Image
103
 
 
9
 
10
  os.system("git clone https://github.com/FrozenBurning/SceneDreamer.git")
11
  sys.path.append("SceneDreamer")
12
+
13
  pretrained_model = dict(file_url='https://drive.google.com/uc?id=1IFu1vNrgF1EaRqPizyEgN_5Vt7Fyg0Mj',
14
  alt_url='', file_size=330571863,
15
  file_path='./scenedreamer_released.pt',)
 
91
  import argparse
92
  from imaginaire.config import Config
93
  from imaginaire.utils.cudnn import init_cudnn
 
 
 
94
  from imaginaire.utils.io import get_checkpoint as get_checkpoint
 
95
  from imaginaire.utils.trainer import \
96
  (get_model_optimizer_and_scheduler, set_random_seed)
 
97
  import gradio as gr
98
  from PIL import Image
99