update

app.py

@@ -6,14 +6,15 @@ import PIL.Image
 import autocuda
 from pyabsa.utils.pyabsa_utils import fprint
 
-# for z_file in findfile.find_cwd_files(and_key=['.zip'],
-#                                       exclude_key=['.ignore', 'git', 'SuperResolutionAnimeDiffusion'],
-#                                       recursive=10):
-#     fprint(f"Extracting {z_file}...")
-#     with zipfile.ZipFile(z_file, 'r') as zip_ref:
-#         zip_ref.extractall(os.path.dirname(z_file))
-
-os.system('unzip random_examples.zip -d random_examples')
+try:
+    for z_file in findfile.find_cwd_files(and_key=['.zip'],
+                                          exclude_key=['.ignore', 'git', 'SuperResolutionAnimeDiffusion'],
+                                          recursive=10):
+        fprint(f"Extracting {z_file}...")
+        with zipfile.ZipFile(z_file, 'r') as zip_ref:
+            zip_ref.extractall(os.path.dirname(z_file))
+except Exception as e:
+    os.system('unzip random_examples.zip')
 
 from diffusers import (
     AutoencoderKL,
@@ -597,4 +598,4 @@ print(f"Space built in {time.time() - start_time:.2f} seconds")
 
 if not is_colab:
     demo.queue(concurrency_count=2)
-demo.launch(debug=is_colab, enable_queue=True, share=is_colab)
+demo.launch(debug=is_colab, enable_queue=True, share=is_colab)

requirements.txt

@@ -4,7 +4,7 @@ torchvision
 autocuda
 findfile
 pyabsa
-git+https://github.com/huggingface/diffusers.git
+diffusers==0.10.0
 scipy
 git+https://github.com/huggingface/transformers.git
 ftfy

test.py

@@ -1,23 +0,0 @@
-# -*- coding: utf-8 -*-
-# file: test.py
-# time: 23:21 2023/1/27
-# author: yangheng <[email protected]>
-# github: https://github.com/yangheng95
-# huggingface: https://huggingface.co/yangheng
-# google scholar: https://scholar.google.com/citations?user=NPq5a_0AAAAJ&hl=en
-# Copyright (C) 2021. All Rights Reserved.
-
-def fixBadZipfile(zipFile):
- f = open(zipFile, 'r+b')
- data = f.read()
- pos = data.find('\x50\x4b\x05\x06') # End of central directory signature
- if (pos > 0):
-     f.seek(pos + 22)   # size of 'ZIP end of central directory record'
-     f.truncate()
-     f.close()
- else:
-     pass
-
-
-if __name__ == '__main__':
-    fixBadZipfile('RealESRGANv030/weights/RealESRGAN_x4plus_anime_6B.zip')