Update app.py

app.py

@@ -11,81 +11,6 @@ os.system("git clone https://github.com/FrozenBurning/SceneDreamer.git")
 os.system("cp -r SceneDreamer/* ./")
 os.system("bash install.sh")
 
-pretrained_model = dict(file_url='https://drive.google.com/uc?id=1IFu1vNrgF1EaRqPizyEgN_5Vt7Fyg0Mj',
-                            alt_url='', file_size=330571863,
-                            file_path='./scenedreamer_released.pt',)
-
-
-def download_file(session, file_spec, use_alt_url=False, chunk_size=128, num_attempts=10):
-    file_path = file_spec['file_path']
-    if use_alt_url:
-        file_url = file_spec['alt_url']
-    else:
-        file_url = file_spec['file_url']
-
-    file_dir = os.path.dirname(file_path)
-    tmp_path = file_path + '.tmp.' + uuid.uuid4().hex
-    if file_dir:
-        os.makedirs(file_dir, exist_ok=True)
-
-    progress_bar = tqdm(total=file_spec['file_size'], unit='B', unit_scale=True)
-    for attempts_left in reversed(range(num_attempts)):
-        data_size = 0
-        progress_bar.reset()
-        try:
-            # Download.
-            data_md5 = hashlib.md5()
-            with session.get(file_url, stream=True) as res:
-                res.raise_for_status()
-                with open(tmp_path, 'wb') as f:
-                    for chunk in res.iter_content(chunk_size=chunk_size<<10):
-                        progress_bar.update(len(chunk))
-                        f.write(chunk)
-                        data_size += len(chunk)
-                        data_md5.update(chunk)
-
-            # Validate.
-            if 'file_size' in file_spec and data_size != file_spec['file_size']:
-                raise IOError('Incorrect file size', file_path)
-            if 'file_md5' in file_spec and data_md5.hexdigest() != file_spec['file_md5']:
-                raise IOError('Incorrect file MD5', file_path)
-            break
-
-        except Exception as e:
-            # print(e)
-            # Last attempt => raise error.
-            if not attempts_left:
-                raise
-
-            # Handle Google Drive virus checker nag.
-            if data_size > 0 and data_size < 8192:
-                with open(tmp_path, 'rb') as f:
-                    data = f.read()
-                links = [html.unescape(link) for link in data.decode('utf-8').split('"') if 'confirm=t' in link]
-                if len(links) == 1:
-                    file_url = requests.compat.urljoin(file_url, links[0])
-                    continue
-
-    progress_bar.close()
-
-    # Rename temp file to the correct name.
-    os.replace(tmp_path, file_path) # atomic
-
-    # Attempt to clean up any leftover temps.
-    for filename in glob.glob(file_path + '.tmp.*'):
-        try:
-            os.remove(filename)
-        except:
-            pass
-
-print('Downloading SceneDreamer pretrained model...')
-with requests.Session() as session:
-    try:
-        download_file(session, pretrained_model)
-    except:
-        print('Google Drive download failed.\n')
-
-
 
 import os
 import torch