commit

app.py

@@ -25,7 +25,7 @@ from lxml.html import fromstring
 from diffusers.utils import export_to_video, load_image
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file, save_file
-from diffusers import FluxPipeline, HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
+from diffusers import FluxPipeline, CogVideoXImageToVideoPipeline
 from PIL import Image, ImageDraw, ImageFont
 
 # logging
@@ -126,65 +126,16 @@ function custom(){
 
 image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=torch.float32).to(device)
 
-transformer = HunyuanVideoTransformer3DModel.from_pretrained(
-    "hunyuanvideo-community/HunyuanVideo", subfolder="transformer", torch_dtype=torch.bfloat16
-)
-video_pipe = HunyuanVideoPipeline.from_pretrained(
-    "hunyuanvideo-community/HunyuanVideo", transformer=transformer, torch_dtype=torch.float16
+video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
+    "THUDM/CogVideoX-5b-I2V",
+    torch_dtype=torch.bfloat16
 ).to(device)
 
 video_pipe.vae.enable_tiling()
 video_pipe.vae.enable_slicing()
-video_pipe.enable_model_cpu_offload()
 
 # functionality
 
-def run(cmd):
-    return str(subprocess.run(cmd, shell=True, capture_output=True, env=None).stdout)
-
-def xpath_finder(str,pattern):
-    try:
-        return ""+fromstring(str).xpath(pattern)[0].text_content().lower().strip()
-    except:
-        return ""
-
-def translate(text,lang):
-    if text == None or lang == None:
-        return ""       
-    text = re.sub(f'[{punctuation}]', '', re.sub('[ ]+', ' ', text)).lower().strip()
-    lang = re.sub(f'[{punctuation}]', '', re.sub('[ ]+', ' ', lang)).lower().strip()    
-    if text == "" or lang == "":
-        return ""
-    if len(text) > 38:
-        raise Exception("Translation Error: Too long text!")
-    user_agents = [
-        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36',
-        'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36',
-        'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36',
-        'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15',
-        'Mozilla/5.0 (Macintosh; Intel Mac OS X 13_1) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15'
-    ]
-    text = text.strip()
-    query_text = f'Please translate {text}, into {lang}'
-    url = f'https://www.google.com/search?q={query_text}'
-    print(url)
-    content = str(requests.get(
-        url = url,
-        headers = {
-            'User-Agent': random.choice(user_agents)
-        }
-    ).content)
-    translated = text
-    src_lang = xpath_finder(content,'//*[@class="source-language"]')
-    trgt_lang = xpath_finder(content,'//*[@class="target-language"]')
-    src_text = xpath_finder(content,'//*[@id="tw-source-text"]/*')
-    trgt_text = xpath_finder(content,'//*[@id="tw-target-text"]/*')
-    if trgt_lang == lang:
-        translated = trgt_text
-    ret = re.sub(f'[{punctuation}]', '', re.sub('[ ]+', ' ', translated)).lower().strip()
-    print(ret)
-    return ret
-    
 def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))