Spaces:
Runtime error
Runtime error
FrozenBurning
commited on
Commit
·
ba74ae4
1
Parent(s):
1299110
Update app.py
Browse files
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 |
|